ljfAdmin

你好，我运行CSTrack的训练代码，为什么会出现错误：TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.恕我愚笨，我感觉您的源代码很好，但是为什么会出现这个错误？望指教

5

File "/SOTS-MOT/CSTrack/tracking/train_cstrack.py", line 526, in train(hyp, opt, device, tb_writer) File "/SOTS-MOT/CSTrack/tracking/train_cstrack.py", line 380, in train save_dir=log_dir) File "\SOTS-MOT\CSTrack\tracking\..\lib\core\mot\eval.py", line 197, in test plot_images(img, output_to_target(output, width, height), paths, str(f), names) #...

您好，效果不尽人意？？？？

2

您好，我使用了您的pytorch版本的YOLOv1代码，由于自身电脑限制，batch_size设置为2，相应的减少了学习率一个量级，在VOC2007训练集上训练，然后再VOC2007测试集上测试，按照您的设置的训练50迭代，但是效果不好不到10%的mAP，然后我又先后增加了迭代次数为100，150,但是最终效果都不尽人意，都没有您的教程中的效果，对了，我是从头训练的；此外附加一下，您的代码中的网络的定义，由于padding的设置，你的网络最终的输出尺寸有一定问题；