如何直接拿预训练模型进行预测? --m参数应该填什么?
Created by: Derek-Kun
我看有一个10万分类的预训练模型,下载下来以后里面是一个ResNet50_vd_10w_pretrained文件夹,里面都是weights,mean,variance等文件,下载的其它ResNet50_vd_pretrained预训练模型里面也是这样, 看教程说分类预测框架有三种,下载的哪些预训练模型都没有MODEL文件,所以我猜应该是persistable 模型,可是不管是模型转换然后再预测还是直接 “训练引擎 + persistable 模型预测”,都有一个 “--m=模型名称 “参数,我填了ResNet50_vd_10w_pretrained就报错说KEY错误,里面的介绍是”model_file(简写 m):模型文件路径,如 ./resnet50-vd/model“,可是persistable 模型里面根本就没有MODEL文件,要转换的还还是要输入
python tools/export_model.py
--m=模型名称
--p=persistable 模型路径
--o=model和params保存路径
我真不知道该怎么用预训练模型预测了。有大神来讲一下吗?