robo_urban.pth: robofirefusenet on urban dataset
robo_fire_best.pth: robofirefusenet on wildfire dataset
robo_fire_aug.pth: robofirefusenet on wildfire dataset but with more augmentation (less miou but higher generalization)
'robo_fire_robust.pth': robofirefusenet on wildfire dataset with robust module