No module named gpu_nms

longcw / yolo2-pytorch

YOLOv2 in PyTorch

1.54k stars 421 forks source link

No module named gpu_nms #29

Closed herleeyandi closed 7 years ago

herleeyandi commented 7 years ago

Hello I am new in yolov2, when I just run python demo.py I got this error. Compiling run successfully. Anybody know how to fix this? -Thank you-

herleeyandi commented 7 years ago

Sorry I haven't seen a small error in compilation. Now it works. Actually the problem is the failure in compilation process because my GCC version is 4.7 which should be updated to 4.9 to recognize some command. Really nice works.