运行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
运行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
不知道何原因,求解决方案啊谢谢