https://github.com/superannotateai/active_learning/blob/aa2b9ae325ce84cee8dbf27e3b7268f824ab89ac/object_detection/pascal_voc_ssd/eval.py#L441 If you want to run the eval.py code for ssd object detection, you need to correct line 441 as detections = net(x).data