DETR训练日志
1. 代码来源
代码源自作者的Github: https://github.com/facebookresearch/detr?tab=readme-ov-file
2. 数据来源
在DETR中只使用了COCO2017数据集,其中训练集有118288张图像,验证集有5001张数据,测试集有40671张数据,训练集、验证集以及测试集的比例是72% : 3% : 25%。标签格式是.json格式。
3. 环境配置
pytorch安装:我的cuda版本是11.8,代码的官网要求pytorch是1.5+,我选择的是2.0.0版本的pytorch.
#安装命令如下
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0