模型评估
September 9, 2022 · View on GitHub
评估测试集
- 确认
Awesome-Backbones/datas/annotations.txt标签准备完毕 - 确认
Awesome-Backbones/datas/下test.txt与annotations.txt对应 - 在
Awesome-Backbones/models/下找到对应配置文件 - 按照
配置文件解释修改参数，主要修改权重路径 - 在
Awesome-Backbones打开终端运行
python tools/evaluation.py models/mobilenet/mobilenet_v3_small.py
单张图像检测
- 在
Awesome-Backbones打开终端运行
python tools/single_test.py datasets/test/dandelion/14283011_3e7452c5b2_n.jpg models/mobilenet/mobilenet_v3_small.py
参数说明：
img : 被测试的单张图像路径
config : 模型配置文件，需注意修改配置文件中data_cfg->test->ckpt的权重路径，将使用该权重进行预测
--classes-map : 数据集对应的标签文件，默认datas/annotations.txt
--device : 推理所用设备，默认GPU
--save-path : 保存路径，默认不保存
批量图像检测
- 在
Awesome-Backbones打开终端运行
python tools/batch_test.py datasets/test/dandelion models/mobilenet/mobilenet_v3_small.py --show
参数说明：
path : 待批量检测的图片文件夹路径
config : 模型配置文件，需注意修改配置文件中data_cfg->test->ckpt的权重路径，将使用该权重进行预测
--classes-map : 数据集对应的标签文件，默认datas/annotations.txt
--device : 推理所用设备，默认GPU
--save-path : 保存路径，默认不保存
--show':批量检测时是否显示图片
视频检测
- 在
Awesome-Backbones打开终端运行
python tools/video_test.py datas/demo.mp4 models/mobilenet/mobilenet_v3_small.py --show
参数说明：
video : 被测试的视频路径
config : 模型配置文件，需注意修改配置文件中data_cfg->test->ckpt的权重路径，将使用该权重进行预测
--classes-map : 数据集对应的标签文件，默认datas/annotations.txt
--device : 推理所用设备，默认GPU
--save-path : 保存路径，默认不保存
--show : 是否开启可视化预览