From a56942686b09137cb0289cea8753720a37e0ea00 Mon Sep 17 00:00:00 2001 From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com> Date: Thu, 14 Jan 2021 13:49:29 +0800 Subject: [PATCH] Update yolo3.py --- nets/yolo3.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nets/yolo3.py b/nets/yolo3.py index 6916db5..2b8762f 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) -- GitLab