search

YonghaoHe / LFD-A-Light-and-Fast-Detector

LFD is a big update upon LFFD. Generally, LFD is a multi-class object detector characterized by lightweight, low inference latency and superior precision. It is for real-world appilcations.
418 stars 82 forks source link

errors when run ~/WIDERFACE_train/predict.py #39

Open zkxwy1996 opened 3 years ago

zkxwy1996 commented 3 years ago

pytorch=1.7.1 cuda=11.2 GeForce RTX 3080 when i run ~/WIDERFACE_train/predict.py, the error occurs:

File "predict.py", line 22, in results = config_dict['model'].predict_for_single_image(image, aug_pipeline=simple_widerface_val_pipeline, classification_threshold=0.5, nms_threshold=0.3) File "/root/zkx/LFD-A-Light-and-Fast-Detector-master/lfd/model/lfd.py", line 634, in predict_for_single_image nms_bboxes, nms_labels = multiclass_nms( File "/root/zkx/LFD-A-Light-and-Fast-Detector-master/lfd/model/utils/nms.py", line 214, in multiclass_nms dets, keep = batched_nms(bboxes, scores, labels, nms_cfg) File "/root/zkx/LFD-A-Light-and-Fast-Detector-master/lfd/model/utils/nms.py", line 153, in batched_nms nms_bboxes, kept_indexes = nms_op(torch.cat([bboxes_for_nms, scores[:, None]], -1), **nmscfg) File "/root/zkx/LFD-A-Light-and-Fast-Detector-master/lfd/model/utils/nms.py", line 53, in nms inds = nms_ext.nms(dets_th, iou_thr) RuntimeError: CUDA error: no kernel image is available for execution on the device

Could u help? Thank you very much!

YonghaoHe commented 2 years ago

@zkxwy1996 do you install LFD successfully ? It seems that nms go wrong. You have to make sure that nms is compiled correctly.