longcw / MOTDT

Real-time Multiple People Tracking with Deeply Learned Candidate Selection and Person Re-identification
MIT License
531 stars 130 forks source link

你好,我在运行eval_mot.py时出错 : cudnn_status_execution_failed,具体报错如下,急求解决方案啊谢谢 #25

Open feixiaoyu2223 opened 5 years ago

feixiaoyu2223 commented 5 years ago

运行python eval_mot.py: 2019-04-01 16:20:54 [INFO]: start seq: MOT16-02 loading layers from squeezenet1_1... load cls model from: data/squeezenet_small40_coco_mot16_ckpt_10.h5 2019-04-01 16:20:59 [INFO]: Load ReID model from data/googlenet_part8_all_xavier_ckpt_56.h5 2019-04-01 16:20:59 [INFO]: Processing frame 0 (100000.00 fps) Traceback (most recent call last): File "eval_mot.py", line 158, in show_image=False) File "eval_mot.py", line 105, in main save_dir=output_dir, show_image=show_image) File "eval_mot.py", line 58, in eval_seq online_targets = tracker.update(frame, det_tlwhs, None) File "/home/wr506/Documents/flyfly/MOTDT/tracker/mot_tracker.py", line 222, in update self.classifier.update(image) File "/home/wr506/Documents/flyfly/MOTDT/models/classification/classifier.py", line 87, in update self.score_map = self.model(im_var) File "/home/wr506/anaconda3/envs/fly/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, kwargs) File "/home/wr506/Documents/flyfly/MOTDT/models/classification/rfcn_cls.py", line 103, in forward feats = self.feature_extractor(x) File "/home/wr506/anaconda3/envs/fly/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, *kwargs) File "/home/wr506/Documents/flyfly/MOTDT/models/backbone/sqeezenet.py", line 59, in forward x2 = self.conv1(x) File "/home/wr506/anaconda3/envs/fly/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, kwargs) File "/home/wr506/anaconda3/envs/fly/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward input = module(input) File "/home/wr506/anaconda3/envs/fly/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, **kwargs) File "/home/wr506/anaconda3/envs/fly/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: CUDNN_STATUS_EXECUTION_FAILED

不知道何原因,求解决方案啊谢谢