Skip to content

Latest commit

 

History

History
71 lines (48 loc) · 2.32 KB

How_to_eval.md

File metadata and controls

71 lines (48 loc) · 2.32 KB

模型评估

BILIBILI

评估测试集

  • 确认Awesome-Backbones/datas/annotations.txt标签准备完毕
  • 确认Awesome-Backbones/datas/test.txtannotations.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 : 保存路径,默认不保存

results

批量图像检测

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