소스 검색

add fcos backbone

yjh0410 10 달 전
부모
커밋
ea9fc23188
2개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 1
      yolo/models/fcos/fcos_backbone.py
  2. 0 1
      yolo/models/yolov5/modules.py

+ 1 - 1
yolo/models/fcos/fcos_backbone.py

@@ -7,7 +7,7 @@ except:
     from  resnet import build_resnet
 
 
-# --------------------- Yolov1's Backbone -----------------------
+# --------------------- FCOS's Backbone -----------------------
 class FcosBackbone(nn.Module):
     def __init__(self, cfg):
         super().__init__()

+ 0 - 1
yolo/models/yolov5/modules.py

@@ -73,4 +73,3 @@ class CSPBlock(nn.Module):
         out = self.conv_layer_3(torch.cat([x1, x2], dim=1))
 
         return out
-