DAMO-YOLO是阿里巴巴达摩院最近新推出的轻量级目标检测框架。作为第一批使用的同学,这里对如何使用DAMO-YOLO进行第三方数据训练和finetune做一个简单的介绍。

一、下载安装DAMO-YOLO

按照官网提供的教程,逐步执行即可。这里由于我们使用的Linux服务器,因此cocoapi选择了Linux版本的命令。

git clone https://github.com/tinyvision/damo-yolo.git
cd DAMO-YOLO/
conda create -n DAMO-YOLO python=3.7 -y
conda activate DAMO-YOLO
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
pip install -r requirements.txt
export PYTHONPATH=$PWD:$PYTHONPATHpip install cython;
pip install git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI # for Linux

二、准备训练数据

这里我使用的是私有数据,按照官网教程提示(DAMO-YOLO/CustomDatasetTutorial.md at master · tinyvision/DAMO-YOLO · GitHub),将标注文件转成coco格式后存在了目录/home/cwhgn/datasets/boxy/annotations/下面,名字分别为boxy_coco_train.json,boxy_coco_valid.json和boxy_coco_test.json。对应图片都放在了/home/cwhgn/datasets/boxy/data下面。

三、将训练数据关联到DAMO-YOLO

利用下面命令,将准备好的数据关联到damo-yolo里。

ln -s /home/cwhgn/datasets/boxy datasets/boxy

打开damo/config/paths_catalog.py文件,将coco_2017_train/coco_2017_val/coco_2017_test_dev替换成我们自己数据的路径,如下图所示:

四、修改训练配置文件

准备好数据后,我们就可以根据需要修改训练配置文件了。如果是finetune的话,需要额外下载DAMO-YOLO官网提供的训练好的模型。这里我们准备使用DAMO-YOLO-S*作为初始化模型进行finetune。直接将下载好的damoyolo_tinynasL25_S.pth存在了DAMO-YOLO目录下面。

打开DAMO-YOLO-S训练配置文件configs/damoyolo_tinynasL25_S.py。一方面,将初始化模型加入到配置文件中。另一方面,将检测的类别数修改成我们训练数据一直。我的数据只有一类,因此改成了1。如下图所示:

五、进行训练

至此,我们就可以用以下命令愉快的跑起来我们的训练啦,训练截图如下图所示。

python -m torch.distributed.launch --nproc_per_node=8 tools/train.py -f configs/damoyolo_tinynasL25_S.py

DAMO-YOLO第三方数据训练教程相关推荐

  1. 用$.getJSON() 和$.post()获取第三方数据做页面 ——惠品折页面(1)

    用$.getJSON() 和$.post()获取第三方数据做页面 首页 index.html 页面 需要jquery  和 template-web  js文件   可以直接在官网下载 中间导航条的固 ...

  2. 如何利用第三方数据进行大数据分析

    企业如何避免迷失在数据网络中?通过掌握大数据和第三方数据.但是,掌握第三方数据可能会非常具有挑战性,称其遵守严格的数据隐私规范是不可错过的.数据在增长,可以利用的机会也在增加.为了从第三方数据孤岛中获 ...

  3. 如何对第三方数据进行大数据分析

    企业如何避免迷失在数据网络中?通过掌握大数据和第三方数据.然而,掌握第三方数据可能会非常具有挑战性,称其遵守严格的数据隐私规范方面,它是不可错过的.数据正在增长,可利用的机会也在增加.为了从第三方数据 ...

  4. 如何快速实现西门子S7-200/300 PLC转Modbus-TCP协议与第三方数据对接

    如何快速实现西门子S7-200/300 PLC转Modbus-TCP协议与第三方数据对接 引言 西门子 SIMATIC 自动化控制系统在工业控制市场应用相当广泛,凭借其安全可靠性.全集成产品线和优异的 ...

  5. 第一方数据,第二方数据,第三方数据,都是什么意思?

    数据无处不在--我们每天创造超过2.5艾字节(exabytes)的数据,研究人员预测到2025年将会有163泽字节(zettabytes)的数据.此外,还有许多种类的数据与企业有关.我们有第一方,第三 ...

  6. ios第三方数据请求 UI_15

    AppDelegate.m //指定根视图self.window.rootViewController = [[[UINavigationController alloc]initWithRootVi ...

  7. 读取EXCEL文件数据,再调用第三方接口,将第三方数据重新写入到EXCEL文件

    读取EXCEL文件数据,再调用第三方接口,将第三方数据重新写入到EXCEL文件 工作中涉及很多提供文档数据,少则几条,多则上万,少的可以自己编辑一个,静态final来自己定义,一旦数太多得话,就得使用 ...

  8. 银行风险预警 第三方数据_网络第三方风险

    银行风险预警 第三方数据 现代软件开发的性质使得很难知道代码的实际编写位置和编写者. (The nature of modern software development makes it hard ...

  9. XDC+荣获2018年度中国IDC产业优秀第三方数据中心奖

    2018年12月11日-13日,第十三届中国IDC产业年度大典在北京举行.中国IDC产业年度大典是中国IDC行业的标志性盛会,年度评选更是每届大会最为醒目的焦点.继获得"优秀数据中心最具创新 ...

最新文章

  1. mysql 断开的管道_ClientAbortException: java.net.SocketException: 断开的管道
  2. AI帮你写Python,安装只需5步,还能任你调教 | 开源
  3. Linux 操作命令(1)
  4. 跟前腾讯总监学Java实战项目
  5. mysql 8.0 创建函数_MySQL 8.0 新增特性
  6. python 线程池回收_python实现线程池
  7. AIX samba服务器配置
  8. redis延迟队列 实现_灵感来袭,基于Redis的分布式延迟队列
  9. lnmp—MemCache的作用
  10. 一夜抢空880万!中国最狠印钞机,终于开始收割年轻人了
  11. php zip class,DedeCMS 自带压缩工具 zip.class.php 的用法
  12. nodejs template
  13. Java中String,StringBuffer,StringBuilder的区别
  14. python os读取文件
  15. java 替换pdf 文本_java 查找替换pdf中的指定文本
  16. 由我国科学家研制的计算机,由我国科学家研制的系列超级计算机综合技术处于国际领先水平,2015年11月,全球超级计算机...
  17. (II)第二节:HelloWordl 案例
  18. API接口——睡眠带开放能力
  19. 记录,太坑了,深度学习显卡损坏原来是这样子的。
  20. 20款常用的在线配色工具

热门文章

  1. HTML超链接怎么设置下拉菜单,html超链接怎么设置
  2. 让计算机猜测你心中想的0-100之间的一个整数
  3. linux scons源码安装,SCons 第一章 构建并安装SCons
  4. 边缘计算的100个术语
  5. Apache HBase
  6. js编写一个函数,计算任意两个数字之间所能组成的奇数个数,数字必须是个位数。...
  7. 网站在线监控工具Statping
  8. PDPS软件:机器人焊枪工具自动选取功能介绍与使用方法
  9. 系统工程理论与实践投稿经验_【系统工程理论与实践杂志】投稿经验与期刊点评_审稿、版面费用_退稿_要求与流程_影响因子_级别_发行周期_见刊时间_极作期刊...
  10. Html+Tomcat+Http协议介绍