We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dfd0a92 commit d4a5354Copy full SHA for d4a5354
mmdet/apis/inference.py
@@ -110,8 +110,8 @@ def inference_detector(model, img):
110
data = test_pipeline(data)
111
data = collate([data], samples_per_gpu=1)
112
# just get the actual data from DataContainer
113
- data['img_metas'] = data['img_metas'][0].data
114
- data['img'] = data['img'][0].data
+ data['img_metas'] = [i.data[0] for i in data['img_metas']]
+ data['img'] = [i.data[0] for i in data['img']]
115
if next(model.parameters()).is_cuda:
116
# scatter to specified GPU
117
data = scatter(data, [device])[0]
0 commit comments