search

Zhongdao / Towards-Realtime-MOT

Joint Detection and Embedding for fast multi-object tracking
MIT License
2.38k stars 539 forks source link

demo error #142

Open nicholaswhf opened 4 years ago

nicholaswhf commented 4 years ago

I got the following error Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

when executed at with torch.no_grad(): pred = self.model(im_blob) at multitrackers.py-JDETracker-update()

but i deleted the '.cuda()' and edited the init() as

        device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
        self.model = Darknet(opt.cfg).to(device)
        # load_darknet_weights(self.model, opt.weights)
        self.model.load_state_dict(torch.load(opt.weights, map_location='cpu')['model'], strict=False)
        #self.model.cuda().eval()