DINO-ubuntu20.04-单机6卡/3卡训练自己的数据集
文章目录
前言
环境:ubuntu=20.04, python=3.7.3, torch=1.9.0, cuda=11.1
一、代码及数据准备
1、将自己数据转为-coco格式数据
COCODIR/
├── train2017/
├── val2017/
└── annotations/
├── instances_train2017.json
└── instances_val2017.json
2、下载代码
git clone https://github.com/IDEA-Research/DINO.git
cd DINO