SeaDronesSee数据集部署

SeaDronesSee数据集为无人机拍摄的海洋救援相关的数据集。目标框有’ignored’, ‘swimmer’, ‘boat’, ‘jetski’, ‘life_saving_appliances’, 'buoy’六个类别。

原始的官网dang下来的数据集是由Java标注的json格式的数据集，无法直接在YOLO上面部署，需要进行数据格式转换，将json格式转换成txt格式。下面进行教程：

一、数据预处理

1. 将解压后的Compressed Version中的images和annotations粘贴到data中
2. 运行data_process.py。将会生成三个txt文件和一个labels文件夹
3. 把images/test、train、val中的所有图片移动到images下

   import os
   from tqdm import tqdm
   from PIL import Image
   import json
   
   def generate_txt_file(folder_path, txt_file_path):
       print("正在处理文件夹:", folder_path)
       with open(txt_file_path, 'w') as txt_file:
           file_list = os.listdir(folder_path)
           for file_name in file_list:
               if file_name.endswith('.jpg') or file_name.endswith('.png'):
                   txt_file.write('data/images/' + file_name + '\n')
       print("已生成文件:", txt_file_path)
   
   
   def JSONtoTXT(mode,