下面这张图是修改为 HelmetDataset 的内容: 在创建完后,我们还需要去 mmdet/datasets/__init__.py 文件里进行录入,.helmet 指的就是我们的 helmet.py 文件,我们也可以输入完整的路径,import 的内容就是我们自定义的 dataset 名字。最后在下面也要加上对应 Dataset 的名称。 配置文件准备及修改 由于在 MMDetecti...
首先需要准备好数据集的标注文件以及 config 文件,确保能够被 dataset 所读取(可以通过tools/misc/browse_dataset.py 工具进行验证),然后需要确保环境中安装了 scipy,然后运行命令: 代码语言:javascript 代码运行次数:0 运行 AI代码解释 python tools/analysis_tools/optimize_anchors.py ${CONFIG}--algorithm k-means-...
本节主要说明下mmdetection-mini到目前为止已经新增的一些用来分析数据,分析模型,分析模型预测质量的工具,在实际项目中针对自己的数据进行可视化分析非常有用。例如感受野计算、特征图分析、loss函数分析、推理时间分析等等。 1. 数据浏览 browse_dataset 给你一个新的目标检测项目,转化为coco格式,设置好cfg后,难道不需要...
我想作为一个有经验的工程师必然少不了这个步骤。 故browse_dataset可以对datasets吐出的数据进行可视化检查,看下是否有错误。这个工具我是直接从mmdetection里面copy过来的,并修复了在voc那种数据的配置上面出错的bug。 用法非常简单,只需要传入cfg文件即可,以coco数据为例,如下所示” 可视化效果看起来比较难看,不过我懒...
我之前都是用labelme直接看的,当然就显得b格不太够,mmdetection里面也可以直接看,用的browse_dataset.py,这个可以看训练集的,当然如果要看测试集的话,可以将config文件里面的data这个dict里面的train的dict中的ann_file和img_profix路径改成和val那个dict中的一样就好了 ...
数据浏览工具 tools/browse_dataset 笔记(持续更新) 第一篇:mmdetection最小复刻版(一):整体概览 或者知乎文章 第二篇:mmdetection最小复刻版(二):RetinaNet和YoloV3分析 或者知乎文章 第三篇:mmdetection最小复刻版(三):神兵利器 或者知乎文章 第四篇:mmdetection最小复刻版(四):独家yolo转化内幕 ...
首先需要准备好数据集的标注文件以及 config 文件,确保能够被 dataset 所读取(可以通过tools/misc/browse_dataset.py工具进行验证),然后需要确保环境中安装了 scipy,然后运行命令: python tools/analysis_tools/optimize_anchors.py ${CONFIG} --algorithm k-means --input-shape ${INPUT_SHAPE [WIDTH HEIGHT]} --...
python tools/misc/browse_dataset.py ${ CONFIG } 四:检测单张图片时用到以下方法: model = init_detector(config_file,checkpoint_file) result=inference_detector(model,img) model.show_result(img, result, model.CLASSES, out_file='testOut.jpg')...
tools/misc/browse_dataset.py ${CONFIG_FILE}--task ${TASK}--output-dir ${OUTPUT_DIR}[--online] 可选参数 online:事实显示结果。需要图形化界面支持,同时需要安装open3d==0.9.0.0 示例: 代码语言:javascript 代码运行次数:0 运行 AI代码解释
在此我们选择使用的模型配置文件是configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py(MMDetection 的名称规范为[model]_(model setting)_[backbone]_[neck]_(norm setting)_(misc)_(gpu x batch)_[schedule]_[dataset].py),在具体使用前,还需要下载该模型对应训练好的模型文件,在模型配置文件对应目录fast...