未验证 提交 a5694268 编写于 作者: B Bubbliiiing 提交者: GitHub

Update yolo3.py

上级 75a8d294
...@@ -38,7 +38,7 @@ class YoloBody(nn.Module): ...@@ -38,7 +38,7 @@ class YoloBody(nn.Module):
#---------------------------------------------------# #---------------------------------------------------#
# 生成darknet53的主干模型 # 生成darknet53的主干模型
# 获得三个有效特征层,他们的shape分别是: # 获得三个有效特征层,他们的shape分别是:
# 13,13,256 # 52,52,256
# 26,26,512 # 26,26,512
# 13,13,1024 # 13,13,1024
#---------------------------------------------------# #---------------------------------------------------#
...@@ -74,7 +74,7 @@ class YoloBody(nn.Module): ...@@ -74,7 +74,7 @@ class YoloBody(nn.Module):
return layer_in, out_branch return layer_in, out_branch
#---------------------------------------------------# #---------------------------------------------------#
# 获得三个有效特征层,他们的shape分别是: # 获得三个有效特征层,他们的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) x2, x1, x0 = self.backbone(x)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册