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:$PYTHONPATH
pip install cython;
pip install git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI # for Linux
二、准备训练数据
这里我使用的是私有数据,按照官网教程提示(