model/backbone/densenet.py

from torch import nn
from torchvision.models.detection.backbone_utils import _resnet_fpn_extractor, resnet_fpn_backbone, BackboneWithFPN

class DenseNet(nn.Module):
    def __init__(self, num_classes):     # @Thuan: Add argumentation in __init__
        super(DenseNet, self).__init__()
        # @Thuan: Initiation model here

        # End of model initiation

    def forward(self, x):
        # @Thuan: Feedforward step here

        return x    # End of feedforward function