Category Archives: 机器学习

机器学习 / 数据挖掘

NanoDet目标检测之"微调训练"

《NanoDet目标检测之"搭建预测篇"》 中,我们搭建了NanoDet的环境,并用默认权重做了简单的预测,本节我们继续用自己的数据做微调。
1 准备数据
NanoDet支持yolo或者coco格式的标注,自行准备即可,我这里以yolo为例,目录结构如下:

TODO
xxx
2 配置
我们需要融合下两个配置文件,复制:
cp nanodet-plus-m_320.yml nanodet-plus-m_320_face.yml
修改1,保存位置:
save_dir: wo[......]

继续阅读

NanoDet目标检测之"搭建预测篇"

NanoDet是一个基于ShuffleNetV2的轻量级的目标检测模型,配合ncnn框架加速后,在中端Android机型能做到20fps+
1 安装环境
为了避免冲突,需要安装好conda,Python 3.8+

安装依赖包
git clone https://github.com/RangiLyu/nanodet.git
cd nanodet
pip install -r requirements.txta
安装代码
python setup.py develop
2 下[......]

继续阅读