- 确认
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
: 是否开启可视化预览