diff --git a/nets/yolo3.py b/nets/yolo3.py index 6916db5e97dcf73dc4a2cc7063df1c1b59bd5c2d..2b8762f262ab252c776a4f46ee80fbdf379c25d9 100644 --- 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)