Update yolo3.py

a5694268 · Bubbliiiing · GitHub · 75a8d294 · a5694268
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

nets/yolo3.py nets/yolo3.py +2 -2

未找到文件。
--- a/nets/yolo3.py
+++ b/nets/yolo3.py
@@ -38,7 +38,7 @@ class YoloBody(nn.Module):
        #---------------------------------------------------#   
        #   生成darknet53的主干模型
        #   获得三个有效特征层，他们的shape分别是：
-        #   13,13,256
+        #   52,52,256
        #   26,26,512
        #   13,13,1024
        #---------------------------------------------------#
@@ -74,7 +74,7 @@ class YoloBody(nn.Module):
            return layer_in, out_branch
        #---------------------------------------------------#   
        #   获得三个有效特征层，他们的shape分别是：
-        #   13,13,256；26,26,512；13,13,1024
+        #   52,52,256；26,26,512；13,13,1024
        #---------------------------------------------------#
        x2, x1, x0 = self.backbone(x)