daodaofr / AlignPS

Code for CVPR 2021 paper: Anchor-Free Person Search
Apache License 2.0
167 stars 34 forks source link

run_test_prw.sh TypeError: 'DataContainer' object is not subscriptable #29

Closed FeboReigns closed 2 years ago

FeboReigns commented 2 years ago

__run_testprw.sh but cmd shows: TypeError: 'DataContainer' object is not subscriptable_

my env: torch-1.8.1+cu111 torchaudio-0.8.1 torchvision-0.9.1+cu111 mmcv-full=1.1.5 more info in cmd: (pytorch_hbs) amax@amax:~/Desktop/personsearch/AlignPS$ sh run_test_prw.sh

:228: RuntimeWarning: scipy._lib.messagestream.MessageStream size changed, may indicate binary incompatibility. Expected 56 from C header, got 64 from PyObject loading annotations into memory... Done (t=0.12s) creating index... index created! /home/amax/anaconda3/envs/pytorch_hbs/lib/python3.9/site-packages/mmcv/cnn/bricks/conv_module.py:100: UserWarning: ConvModule has norm and bias at the same time warnings.warn('ConvModule has norm and bias at the same time') [ ] 0/6112, elapsed: 0s, ETA::228: RuntimeWarning: scipy._lib.messagestream.MessageStream size changed, may indicate binary incompatibility. Expected 56 from C header, got 64 from PyObject :228: RuntimeWarning: scipy._lib.messagestream.MessageStream size changed, may indicate binary incompatibility. Expected 56 from C header, got 64 from PyObject :228: RuntimeWarning: scipy._lib.messagestream.MessageStream size changed, may indicate binary incompatibility. Expected 56 from C header, got 64 from PyObject :228: RuntimeWarning: scipy._lib.messagestream.MessageStream size changed, may indicate binary incompatibility. Expected 56 from C header, got 64 from PyObject /home/amax/anaconda3/envs/pytorch_hbs/lib/python3.9/site-packages/torch/nn/functional.py:3328: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead. warnings.warn("nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.") /home/amax/anaconda3/envs/pytorch_hbs/lib/python3.9/site-packages/torch/nn/functional.py:3454: UserWarning: Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details. warnings.warn( Traceback (most recent call last): File "/home/amax/Desktop/personsearch/AlignPS/./tools/test.py", line 226, in main() File "/home/amax/Desktop/personsearch/AlignPS/./tools/test.py", line 186, in main outputs = multi_gpu_test(model, data_loader, args.tmpdir, File "/home/amax/Desktop/personsearch/AlignPS/mmdet/apis/test.py", line 98, in multi_gpu_test result = model(return_loss=False, rescale=True, **data) File "/home/amax/anaconda3/envs/pytorch_hbs/lib/python3.9/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/amax/anaconda3/envs/pytorch_hbs/lib/python3.9/site-packages/torch/nn/parallel/distributed.py", line 705, in forward output = self.module(*inputs[0], **kwargs[0]) File "/home/amax/anaconda3/envs/pytorch_hbs/lib/python3.9/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/amax/Desktop/personsearch/AlignPS/mmdet/core/fp16/decorators.py", line 51, in new_func return old_func(*args, **kwargs) File "/home/amax/Desktop/personsearch/AlignPS/mmdet/models/detectors/base.py", line 170, in forward return self.forward_test(img, img_metas, **kwargs) File "/home/amax/Desktop/personsearch/AlignPS/mmdet/models/detectors/base.py", line 147, in forward_test return self.simple_test(imgs[0], img_metas[0], **kwargs) File "/home/amax/Desktop/personsearch/AlignPS/mmdet/models/detectors/single_stage_reid.py", line 117, in simple_test bbox_list = self.bbox_head.get_bboxes( File "/home/amax/Desktop/personsearch/AlignPS/mmdet/core/fp16/decorators.py", line 131, in new_func return old_func(*args, **kwargs) File "/home/amax/Desktop/personsearch/AlignPS/mmdet/models/dense_heads/fcos_reid_head_focal_oim_sub.py", line 456, in get_bboxes img_shape = img_metas[img_id]['img_shape'] **TypeError: 'DataContainer' object is not subscriptable**
crsm424 commented 2 years ago

Hi, how did you solve this problem