rowanz / neural-motifs

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

minor TypeError fix #73

Open maximilianmozes opened 5 years ago

maximilianmozes commented 5 years ago

During validation, ObjectDetector.nms_boxes returns None if “nothing was detected“ (line 402 in lib/object_detector.py). This yields a TypeError in ObjectDetector.forward when unrolling the returned objects (line 308 in lib/object_detector.py).

GreenTeaHua commented 5 years ago

thanks, but after your fix, in res_model.py, meet this error. if result.is_none(): return ValueError("heck")