OBBDetection icon indicating copy to clipboard operation
OBBDetection copied to clipboard

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

Open Yangjie0610 opened this issue 4 years ago • 5 comments

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

Yangjie0610 avatar Dec 05 '21 08:12 Yangjie0610

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

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

jbwang1997 avatar Dec 05 '21 08:12 jbwang1997

好的,谢谢

Yangjie0610 avatar Dec 05 '21 09:12 Yangjie0610

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

yangcyz avatar Dec 05 '21 11:12 yangcyz

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

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

Yangjie0610 avatar Dec 06 '21 07:12 Yangjie0610

你好,我配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 都出现了错误

wxggzz avatar Jan 06 '22 02:01 wxggzz