justchenhao / STANet

official implementation of the spatial-temporal attention neural network (STANet) for remote sensing image change detection
BSD 2-Clause "Simplified" License
405 stars 110 forks source link

请问怎么才能得到论文中的F1分数指标呢? #75

Open Ricco567 opened 3 years ago

Ricco567 commented 3 years ago

我现在只能复现到0.864,然后我是先把所有数据集裁剪成256×256,然后再运行python ./train.py --save_epoch_freq 1 --angle 15 --dataroot ./LEVIR-CD0(crop)/train --val_dataroot ./LEVIR-CD0(crop)/val --name LEVIR-CDFAp5 --lr 0.001 --model CDFA --SA_mode PAM --batch_size 8 --load_size 256 --crop_size 256 --preprocess rotate_and_crop,不知道这样有影响吗?还是说不用自己手动裁剪,直接用crop_size 256 这种命令就能裁剪来呢?

gcg0210 commented 3 years ago

请问,您搞明白这个了吗?手动裁剪和命令裁剪的区别

woleto commented 3 years ago

请问您ds设置是多少呢? ds默认是1,内存直接爆炸,后来改为4。 另外我没有手动裁剪,结果就是PAM模式最好才0.767。