You need to sign in or sign up before continuing.
未验证 提交 ca199f73 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix downsample (#420)

上级 79616f79
...@@ -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.
先完成此消息的编辑!
想要评论请 注册