Zhao-Tian-yi / RSDet

49 stars 6 forks source link

配对数据集的组织形式,mask,的设置及训练用到的配置文件(14th or 15th) #6

Closed liting1018 closed 6 months ago

liting1018 commented 6 months ago

作者大大好: 非常感谢您开源的代码,但我对于您提供的代码有一些疑问。 (1)关于配对数据集的组织形式,我看您代码里设置的dict(type='LoadAnnotations', with_bbox=True, with_mask=True),您这里是用到了mask信息嘛,我不太清楚您这里的mask设置,我这边的情况是设置with_mask=True时会报错,我修改为False能够跑通。 (2)对于FLIR数据集来说,您提供的训练配置文件包括了configs/fusion/RSDet/faster_rcnn_r50_common_unique_FLIR_14th.py和configs/fusion/RSDet/faster_rcnn_r50_common_unique_FLIR_15th.py,请问您论文中的结果是对应哪个配置文件呢?

期待能得到您的回复,感谢感谢!

love-whut commented 5 months ago

作者大大好: 非常感谢您开源的代码,但我对于您提供的代码有一些疑问。 (1)关于配对数据集的组织形式,我看您代码里设置的dict(type='LoadAnnotations', with_bbox=True, with_mask=True),您这里是用到了mask信息嘛,我不太清楚您这里的mask设置,我这边的情况是设置with_mask=True时会报错,我修改为False能够跑通。 (2)对于FLIR数据集来说,您提供的训练配置文件包括了configs/fusion/RSDet/faster_rcnn_r50_common_unique_FLIR_14th.py和configs/fusion/RSDet/faster_rcnn_r50_common_unique_FLIR_15th.py,请问您论文中的结果是对应哪个配置文件呢?

期待能得到您的回复,感谢感谢!

image 请问你遇到过这种情况嘛?

Zhao-Tian-yi commented 5 months ago

(1)mask设置为False就可以,没有使用mask信息,因为我改的时候源代码是这样写的,可能在转换数据集的时候,我使用的脚本生成了mask的参数,所以我没有报错,也就没有注意更改这个参数。(2)您可以使用14th的配置文件,可以达到不错的效果,但本文仍然再审,若被接受后,有时间我会对本库代码进行更新。(3)您的截图可以进行debug找到对应的位置,那里有被注释掉的代码,根据数据集更换一下。

love-whut commented 5 months ago

(1)mask设置为False就可以,没有使用mask信息,因为我改的时候源代码是这样写的,可能在转换数据集的时候,我使用的脚本生成了mask的参数,所以我没有报错,也就没有注意更改这个参数。(2)您可以使用14th的配置文件,可以达到不错的效果,但本文仍然再审,若被接受后,有时间我会对本库代码进行更新。(3)您的截图可以进行debug找到对应的位置,那里有被注释掉的代码,根据数据集更换一下。

您好,我把self.pool替换为FLIR数据集的后报了这个错 image image

love-whut commented 5 months ago

(1)mask设置为False就可以,没有使用mask信息,因为我改的时候源代码是这样写的,可能在转换数据集的时候,我使用的脚本生成了mask的参数,所以我没有报错,也就没有注意更改这个参数。(2)您可以使用14th的配置文件,可以达到不错的效果,但本文仍然再审,若被接受后,有时间我会对本库代码进行更新。(3)您的截图可以进行debug找到对应的位置,那里有被注释掉的代码,根据数据集更换一下。

您好,我把self.pool替换为FLIR数据集的后报了这个错 image image

不好意思,是我类别设置的问题,已经解决了,感谢您的工作和您耐心的回复,希望能看到您更多杰出的工作!

love-whut commented 5 months ago

(1)mask设置为False就可以,没有使用mask信息,因为我改的时候源代码是这样写的,可能在转换数据集的时候,我使用的脚本生成了mask的参数,所以我没有报错,也就没有注意更改这个参数。(2)您可以使用14th的配置文件,可以达到不错的效果,但本文仍然再审,若被接受后,有时间我会对本库代码进行更新。(3)您的截图可以进行debug找到对应的位置,那里有被注释掉的代码,根据数据集更换一下。

另外请问一下修改了kernel_sizeself.IA_fc11self.IA_fc21中的in_features也要做相应的修改嘛? image