Open Maxwell2016LeChouchou opened 11 months ago
模型的权重是通过字典加载的:
pretrained_dict = torch.load(model_path, map_location = device)
pretrained_dict = {k: v for k, v in pretrained_dict.items() if np.shape(model_dict[k]) == np.shape(v)}
官方仓库训练的模型和这个仓库的key不同,你可以对一下两个权重的shape,看看这个仓库的key是什么样的,然后自己改一下官方的key
请问这个您的yolo.py和predict.py可以用旷视训练好的yolox_s的直接做预测推理吗? 我看您的predict.py有可视化heatmap,就想用我在旷视训练好的yolox_s(自己标注的训练集)在您的代码上直接做heatmap可视化。 但是报一大堆backbone的错误,主要是这些: