search

sud0301 / semisup-semseg

136 stars 30 forks source link

save_output_images for cityscapes? #14

Open haiderasad opened 3 years ago

haiderasad commented 3 years ago

if args.save_output_images: if args.dataset == 'pascal_voc': filename = os.path.join(args.save_dir, '{}.png'.format(name[0])) color_file = Image.fromarray(colorize(output).transpose(1, 2, 0), 'RGB') color_file.save(filename) elif args.dataset == 'pascal_context': filename = os.path.join(args.save_dir, filename[0]) scipy.misc.imsave(filename, gt)

why the results of cityscapes evaluation not saved?

haiderasad commented 3 years ago

also kindly tell why we use VOC_30000 for evaluation and not VOC_40000?

sud0301 commented 3 years ago

It is just an example script. You would need to find a suitable checkpoint using a validation set.