关于视频动作识别TSN模型 参数加载的问题
Created by: wwjjy
该代码训练时,如果没有指定预训练的参数,那么它会自动的下载model zoo中预训练的模型,但是该模型应该是基于tsn.txt中如下图配置的方式训练的,此时网络的输入是(3,224,224) 但是如果我们将该配置改为seg_num = 3,seglen = 2,那么网络的输入相应的会变成(6,224,224)。此时会由于卷积的维度和输入的通道数不符报错,错误信息如下: 貌似在程序中没有找到通过哪个参数可以避免掉这种情况发生,只能手动删掉下载并加载预训练参数的部分。 同理在infer时,如果没有指定权重的路径,会自动下载并加载预训练的权重路径,如果使用自定义的数据集,在用户忘了指定权重路径时,应该提示他没有指定路径,而不是使用预训练的权重进行推断,这在排查问题以及测试模型效果时很不友好。