from utils.nms.gpu_nms import gpu_nms

longcw / MOTDT

Real-time Multiple People Tracking with Deeply Learned Candidate Selection and Person Re-identification

MIT License

531 stars 130 forks source link

from utils.nms.gpu_nms import gpu_nms #3

Open hiteyoshi opened 6 years ago

hiteyoshi commented 6 years ago

there is no gpu_nms.py file, should I just use cpu?

hiteyoshi commented 6 years ago

seems like several files lost

mohammed-amr commented 5 years ago

Have you managed to find a fix for this?

JierunChen commented 5 years ago

It works for me to downgrade pytorch1.0.1 with cuda10 to pytorch0.4.1 with cuda9.0. Also make sure the cudatoolkit and nvidia driver installed in the system match with your pytorch version.