Lliu666 / MSHNet

Implement of CVPR2024 'Infrared Small Target Detection with Scale and Location Sensitivity'
41 stars 10 forks source link

新结果的参数可否提供一下,我设定的bs=4,lr=0.05 #7

Open JiangNan-Yang opened 6 months ago

JiangNan-Yang commented 6 months ago

结果是miou66.82 Pd93.54 Fa14.2

yangyahu-1994 commented 5 months ago

你好,你自己训练的模型能够test嘛?

JiangNan-Yang commented 5 months ago

你好,你自己训练的模型能够test嘛?

可以啊,test结果就是上面那个

yangyahu-1994 commented 5 months ago

你好,你自己训练的模型能够test嘛?

可以啊,test结果就是上面那个

训练保存的pkl权重文件好像不能用啊,test的时候

JiangNan-Yang commented 5 months ago

你好,你自己训练的模型能够test嘛?

可以啊,test结果就是上面那个

训练保存的pkl权重文件好像不能用啊,test的时候

weight.pkl这个权重文件可用啊,你可能是文件访问地址错了

yangyahu-1994 commented 5 months ago

你好,你自己训练的模型能够test嘛?

可以啊,test结果就是上面那个

训练保存的pkl权重文件好像不能用啊,test的时候

weight.pkl这个权重文件可用啊,你可能是文件访问地址错了

好的,谢谢。 公开代码好像没有推理可视化结果的代码

yangyahu-1994 commented 5 months ago

你好,你自己训练的模型能够test嘛?

可以啊,test结果就是上面那个

训练保存的pkl权重文件好像不能用啊,test的时候

weight.pkl这个权重文件可用啊,你可能是文件访问地址错了

self.model.load_state_dict(weight['state_dict'])这行代码没有参数['state_dict']就对了

yangyahu-1994 commented 5 months ago

结果是miou66.82 Pd93.54 Fa14.2

我是67.14,93.54,9.49

JiangNan-Yang commented 5 months ago

结果是miou66.82 Pd93.54 Fa14.2

我是67.14,93.54,9.49

有改参数吗?比如bs,lr,什么的

yangyahu-1994 commented 5 months ago

结果是miou66.82 Pd93.54 Fa14.2

我是67.14,93.54,9.49

有改参数吗?比如bs,lr,什么的

没有

yangyahu-1994 commented 5 months ago

结果是miou66.82 Pd93.54 Fa14.2

我是67.14,93.54,9.49

有改参数吗?比如bs,lr,什么的

没有

他这个没有给detec可视化的代码

wikiwwid commented 1 month ago

我跑出67.88,92.86,10.17

Ws1www commented 1 month ago

@wikiwwid 您好,请问NUDT数据集跑出的结果怎样?