fix default type (open-mmlab#5438)

AronLin · web-flow · commit 269bb9e8c6ba · 2021-06-24T18:00:49.000+08:00
diff --git a/mmdet/core/export/pytorch2onnx.py b/mmdet/core/export/pytorch2onnx.py
@@ -153,7 +153,7 @@ def preprocess_example_input(input_config):
         'ori_shape': (H, W, C),
         'pad_shape': (H, W, C),
         'filename': '<demo>.png',
-        'scale_factor': np.ones(4),
+        'scale_factor': np.ones(4, dtype=np.float32),
         'flip': False,
         'show_img': show_img,
     }

Original file line number	Diff line number	Diff line change
`@@ -153,7 +153,7 @@ def preprocess_example_input(input_config):`
`153`	`153`	`'ori_shape': (H, W, C),`
`154`	`154`	`'pad_shape': (H, W, C),`
`155`	`155`	`'filename': '<demo>.png',`
`156`		`- 'scale_factor': np.ones(4),`
	`156`	`+ 'scale_factor': np.ones(4, dtype=np.float32),`
`157`	`157`	`'flip': False,`
`158`	`158`	`'show_img': show_img,`
`159`	`159`	`}`