目标检测 把遥感目标检测RSOD数据集转换为MSCOCO格式

一、RSOD数据集文件组织格式

        RSOD数据集是一个用于遥感图像中目标检测的开放数据集。该数据集包括飞机、油罐、游乐场和立交桥四类目标。数据集的格式是PASCAL VOC格式。

        数据集包括4个文件夹，每个文件夹对应一种目标类型。

1. 飞机数据集（aircraft），包含446幅图像中的4993架飞机。
2. 操场(playground)，包含189幅图像中的191个游乐场。
3. 立交桥(overpass)，包含176幅图像中的180个立交桥。
4. 油罐(oiltank)，包含165幅图像中的1586个油罐​。

        在每一个类别的文件下，包含Annotation和JPEGImages。Annotation下包括label(txt)和xml两个文件夹。这就是RSOD的文件结构。

 

二、RSOD2COCO.py

        以下代码将读取所有类别的所有标注，转换到一个trainval.json中，包含了所有目标信息，没有进行数据集分割。

import json
import os
from PIL import Image

if __name__ == '__main__':
    image_dir = './RSOD/JPEGImages/'
    anno_dir =