基于detectron2框架的深度学习模型载入自定义数据集

一、前言

最近在做微光目标检测的研究工作，使用了Rank_DETR；这个模型是基于detrex框架，而detrex框架又是基于detectron2的。找了一圈没找到载入数据集的地方，后面查阅了资料得知要用API进行注册。

二、步骤

1. 注册数据集：
   在脚本中，我们首先要注册数据集。Detectron2 提供了多种注册数据集的方式，常用的是 register_coco_instances，用于 COCO 格式的数据集。您可以在脚本的开头或配置文件中添加如下代码来注册您的数据集：

   from detectron2.data.datasets import register_coco_instances
   
   register_coco_instances("my_dataset_train", {
        
        }, "path/to/train_annotations.json", "path/to/train_images/")
   register_coco_instances("my_dataset_val", {
        
        }, "path/to/val_annotations.json", "path/to/val_images/")
   

   • "my_dataset_train" 和 "my_dataset_val" 是数据集的名称，您可以按需更改。
   • path/to/train_annotations.json 和 path/to/val_annotations.json 分别是训练和验证数据集的 COCO 格式标注文件路径。
   • path/to/train_images/ 和 path/to/val_images/ 是训练和验证图像的路径。
   <