search

rowanz / neural-motifs

Code for Neural Motifs: Scene Graph Parsing with Global Context (CVPR 2018)
https://rowanzellers.com/neuralmotifs
MIT License
523 stars 114 forks source link

Dimension out of range (expected to be in range of [-1, 0], but got 1) #79

Open aishamina opened 5 years ago

aishamina commented 5 years ago

Traceback (most recent call last): File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 216, in mAp = val_epoch() File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 171, in val_epoch vr.append(val_batch(val_b, batch)) File "/home/lijiayi/neural-motifs-master/models/train_detector.py", line 190, in val_batch result = detector[b] File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 417, in getitem outputs = nn.parallel.parallel_apply(replicas, [batch[i] for i in range(self.num_gpus)]) File "/home/lijiayi/anaconda3/envs/motifs/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 77, in parallel_apply raise output File "/home/lijiayi/anaconda3/envs/motifs/lib/python3.6/site-packages/torch/nn/parallel/parallel_apply.py", line 53, in _worker output = module(*input, *kwargs) File "/home/lijiayi/anaconda3/envs/motifs/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call result = self.forward(input, **kwargs) File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 311, in forward od_box_deltas, im_sizes, File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 395, in nms_boxes thresh=self.thresh, File "/home/lijiayi/neural-motifs-master/lib/object_detector.py", line 445, in filter_det for c_i in valid_cls.squeeze(1).cpu(): RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)