search

bubbliiiing / yolox-pytorch

这是一个yolox-pytorch的源码，可以用于训练自己的模型。
Apache License 2.0
877 stars 184 forks source link

您好，关于预测shape报错的问题。 #157

Open duochifanya opened 4 months ago

duochifanya commented 4 months ago

您好，关于预测shape报错的问题。

检查过多次，

关于model_path： 训练时的model_path = 'model_data/yolox_s.pth'， 预测时的model_path = 我训练后logs里best_epoch_weights.pth

关于classes_path ： classes_path 训练时候 与 预测时候 一致。 classes_path = 'model_data/voc_classes.txt'

训练时的num_classes=1,

voc_classes.txt 是person

还是会报错shepe不一致情况，请问这是为什么呢？

image

duochifanya commented 4 months ago

已解决