2021-10-20

Yolo3报错记录

简单记录一些在试运行yolo时碰到的错误。

1. 内存溢出问题

试验环境为i9-10850k，RTX3080ti，当freeze_batch_size设置超过64，或者unfreeze_batch_size超过超过32时会出现内存溢出问题，此时修改batch_size大小(往小了调)就可以解决。

2. size mismatch for last_layer0.6.weight: copying a param with shape torch.Size(【255, 1024, 1, 1】) from checkpoint, the shape in current model is torch.Size(【20, 1024, 1, 1】).

题主出现此问题是由于预测时使用的分类文件和训练时使用的分类文件不同造成的，即：yolo3根目录下的predict.py文件中：

"classes_path": 'model_data/voc_classes.txt',

需要与根目录下的train.py文件中的：

"classes_path": 'model_data/voc_classes.txt',

保持一致即可。