jbwang1997 / OBBDetection

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

test问题 #106

Open StarBlue98 opened 2 years ago

StarBlue98 commented 2 years ago

我尝试用少量的数据去训练模型(主要为了看看能不能跑通) 已经根据提供的样例训练,代码如下: python tools/train.py configs/obb/oriented_rcnn/faster_rcnn_orpn_r50_fpn_1x_dota10.py

我现在想要用训练好的这个 epoch_12.pth 去测试模型,用了3张dota图片(已经进行剪裁操作),运行代码如下: python tools/test.py configs/obb/oriented_rcnn/faster_rcnn_orpn_r50_fpn_1x_dota10.py epoch_12.pth --format-only

但是一直都没有具体的结果出来,程序一直停在了最后一行,我不知道是不是我的哪个过程操作有问题呢?

return torch.max_pool2d(input, kernel_size, stride, padding, dilation, ceil_mode) [>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>] 30/30, 19.0 task/s, elapsed: 2s, ETA: 0s Merging patch bboxes into full image!!! Multiple processing [ ] 0/3, elapsed: 0s, ETA: (就是这个最后一行) ![Uploading 图片1.png…]()

jbwang1997 commented 2 years ago

如果训练的数据量过少的话,可能导致模型还没有拟合,检测到的目标过多,在结果融合的时候nms不过来

StarBlue98 commented 2 years ago

哈喽,您好,还是和上次类似的问题,这次我使用了15000+的图片(经过剪裁)进行训练,测试时使用25000+的图片(经过剪裁),但是仍然发现无法看到结果,界面停在了这句话:

Multiple processing [ ] 10/25358, 1372.7 task/s, elapsed: 0s, ETA: 18s

我不知道正常的test.py测试的界面显示应该是一个什么样的情况?您可以方便告诉我一下,这个问题该怎么解决呢?非常感谢您的解答!

电脑配置是3090,如下: 图片2

图片1

jbwang1997 commented 2 years ago

哦,如果和图卡住的话可以尝试使用单进程和图。 在--options后面增加nproc=1或直接在这里这里的nproc改为1

StarBlue98 commented 2 years ago

哈喽,元旦快乐,天天开心! 根据您的提示,已经可以顺利测出mAP啦!但是,我还有一点点小疑问嗷:

  1. 我使用的数据集采用的是dota format,那在训练的时候,需不需要注意 x1 y1 这个点指的是 四个角点中最左边的顶点?还是说,保证四个角点,顺时针旋转就可以了呢? (因为我的mAP有点低,还在思考是什么原因?)
  2. 如果是采用 五参数法进行训练, 指的是 长边定义法?还是 OpenCV定义法呢?
  3. 提供的训练代码 是否有在训练过程中使用(或采用了哪些) 数据增强的方法呢?
  4. 我想要用自己的图片训练出一个baseline,不想要对图像进行剪裁,标注格式为dota,那我应该怎么做呢?

感谢您的热情解答,祝身体健康、学习进步!

jbwang1997 commented 2 years ago
  1. 如果是4个点坐标这种目标框格式,不需要注意初始点的位置和方向,只需要注意点之间不要交叉即可。
  2. 5参数是长边定义法,详细可见BboxToolkit对邮箱框的定义.
  3. 原始的模型仅进行水平或垂直方向的翻转,没有使用其他增强方法。
  4. 可以通过继承CustomDataset建立自己的数据集类,只要data_info中box格式与BboxToolkit定义的相同就可以。
StarBlue98 commented 2 years ago

哈喽,我尝试了一下,在参照 mmdet/datasets/obb/dota.py 创建自己的数据集类,但是,最后不会生成相关的 .pkl 文件,所以可以麻烦您具体说一说 该如何操作吗? 这是标注格式 图片1

LowPassFilter-phil commented 2 years ago

你好,我测试出来dota1.0测试出来的map=0?你遇到过这种情况吗