search

jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.
Apache License 2.0
522 stars 112 forks source link

数据集class的定义在哪？？怎么区分dota1.0和1.5？ #83

Open Yangjie0610 opened 2 years ago

Yangjie0610 commented 2 years ago

你好，麻烦问一下， 数据集class的定义在哪(找了一大圈没找到，是我太菜-。-)？？ 怎么区分dota1.0和1.5？难道只是在config中修改num_classes就行了嘛？

jbwang1997 commented 2 years ago

DOTA1.0与1.5的数据格式都是相同的，只需要在BboxToolkit裁图的时候选择相应的config,并改变OBBDetection模型的num_classes就可以训练。

classes的在裁图的时候就有定义，BboxToolkit

Yangjie0610 commented 2 years ago

好的，谢谢

yangcyz commented 2 years ago

你好，我配obb的环境一直没配好，想咨询你一下，你使用的cuda、pytorch都是哪个版本

Yangjie0610 commented 2 years ago

你好，我配obb的环境一直没配好，想咨询你一下，你使用的cuda、pytorch都是哪个版本

我的环境是：cuda11.1，pytorch1.8.0，python3.7，mmcv-full 1.3.8 配置应该没什么大问题吧，我是一次过得，中间没有发现问题

wxggzz commented 2 years ago

你好，我配obb的环境一直没配好，想咨询你一下，你使用的cuda、pytorch都是哪个版本

我的环境是：cuda11.1，pytorch1.8.0，python3.7，mmcv-full 1.3.8 配置应该没什么大问题吧，我是一次过得，中间没有发现问题

你好我的配置和你一样，你正常训练出来了吗。能看一下你的训练命令吗， 我的单卡是 python tools/train.py ./configs/obb/oriented_rcnn/faster_rcnn_orpn_r50_fpn_1x_ms_rr_dota10.py 多卡是 ./tools/dist_train.sh ./configs/obb/oriented_rcnn/faster_rcnn_orpn_r50_fpn_1x_ms_rr_dota10.py 2 都出现了错误