Created by: shippingwang
-
预测一张图脚本 指定 --image_path="XXX" 如python infer.py --model=ResNet50 --pretrained_model="XXX/" --sample_path="XXX.jpeg"
-
预测结果保存json python infer.py --save_json="xxx.json"
-
评估脚本, 结果输出到json文件中
指定 json文件保存地址 --save_json="XXX"
-
模型转换脚本 还是infer.py指定--save_inference来做
-
没有校验集合,只支持训练,训练过程中不评估 指定--validate=False即可
-
模型微调脚本 去除FC层
-
分类: 代码包里写死了acc5的计算,用户设置类别数class_dim小于5的时候会报错
-
支持多卡评估 ,单独评估&边训边测,支持多卡预测