search

JoyeZLearning / DiffDet4SAR

DiffDet4SAR: Diffusion-based Aircraft Target Detection Network for SAR Images(GRSL 2024)
Other
27 stars 3 forks source link

请问如何训练自己的数据集啊？ #9

Closed yangyahu-1994 closed 3 months ago

JoyeZLearning commented 3 months ago

重新建个文件夹dataset放自己的数据集，改一下相应的路径和config里与数据集有关的参数就行

yangyahu-1994 commented 3 months ago

重新建个文件夹dataset放自己的数据集，改一下相应的路径和config里与数据集有关的参数就行

image 我就是按照您的readme做的，但是不行啊。请问config需要修改那几个文件？

JoyeZLearning commented 3 months ago

1、modifying DiffusionDet-main/detectron2/engine/defaults.py and the 98-122 line to your root.

2、像制作coco格式的数据集一样，按照coco.py里的代码，根据自己数据集的类别和均值方差这些，写进入，具体的可以搜，如何使用coco格式训练自己的数据集。 这个过程就是，需要知道模型想学到哪些，就告诉他哪些（类别、数据集的均值方差、图像大小以及训练集和测试集的位置等） 大的detector基本不用动，属于model类，要改的就是和自己的dataset相关的。按照这个思路，可以仔细思考并查看代码，或者进行调试，看一下model对dataset的处理，就知道自己的数据应该如何训练了。

yangyahu-1994 commented 3 months ago

1、modifying DiffusionDet-main/detectron2/engine/defaults.py and the 98-122 line to your root.

2、像制作coco格式的数据集一样，按照coco.py里的代码，根据自己数据集的类别和均值方差这些，写进入，具体的可以搜，如何使用coco格式训练自己的数据集。 这个过程就是，需要知道模型想学到哪些，就告诉他哪些（类别、数据集的均值方差、图像大小以及训练集和测试集的位置等） 大的detector基本不用动，属于model类，要改的就是和自己的dataset相关的。按照这个思路，可以仔细思考并查看代码，或者进行调试，看一下model对dataset的处理，就知道自己的数据应该如何训练了。

数据集就是按照您给的代码制作的，应该没问题。 98-122 line这个也改了，就是报错 image

yangyahu-1994 commented 3 months ago

1、modifying DiffusionDet-main/detectron2/engine/defaults.py and the 98-122 line to your root.

2、像制作coco格式的数据集一样，按照coco.py里的代码，根据自己数据集的类别和均值方差这些，写进入，具体的可以搜，如何使用coco格式训练自己的数据集。 这个过程就是，需要知道模型想学到哪些，就告诉他哪些（类别、数据集的均值方差、图像大小以及训练集和测试集的位置等） 大的detector基本不用动，属于model类，要改的就是和自己的dataset相关的。按照这个思路，可以仔细思考并查看代码，或者进行调试，看一下model对dataset的处理，就知道自己的数据应该如何训练了。

数据集就是按照voc2coco.py制作的