search

vietnh1009 / Yolo-v2-pytorch

YOLO for object detection tasks
MIT License
373 stars 107 forks source link

triu not implemented for Bool #18

Open Asidipta opened 4 years ago

Asidipta commented 4 years ago

conflicting = (ious > nms_threshold).triu(1)

Any work around?