未验证 提交 6e84805b 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix downsample (#421)

上级 a3eb2de1
...@@ -121,7 +121,7 @@ class YOLOv3(object): ...@@ -121,7 +121,7 @@ class YOLOv3(object):
-2] // downsample if image_shape[-2] else None -2] // downsample if image_shape[-2] else None
targets_def[k]['shape'][4] = image_shape[ targets_def[k]['shape'][4] = image_shape[
-1] // downsample if image_shape[-1] else None -1] // downsample if image_shape[-1] else None
downsample // 2 downsample //= 2
inputs_def.update(targets_def) inputs_def.update(targets_def)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册