BHRL icon indicating copy to clipboard operation
BHRL copied to clipboard

onnx conversion error

Open ZXW-King opened this issue 1 year ago • 0 comments

Use your pytorch2onnx.py to convert onnx and the error is as follows: File "/media/xin/work1/github_pro/BHRL/mmdet/models/detectors/base.py", line 168, in forward return self.onnx_export(img[0], img_metas[0]) File "/media/xin/work1/github_pro/BHRL/mmdet/models/detectors/two_stage.py", line 194, in onnx_export x = self.extract_feat(img) File "/media/xin/work1/github_pro/BHRL/mmdet/models/detectors/bhrl.py", line 49, in extract_feat rf_feat = img[1] IndexError: index 1 is out of bounds for dimension 0 with size 1 Please give me an answer, thank you

ZXW-King avatar Feb 02 '24 06:02 ZXW-King