INNER CODE UNIT · Python

GSAM_mask

Shilin-LU/MACE · data_preparation.py:56

                GSAM_mask = (GSAM_mask.to(torch.uint8) * 255).squeeze()
                save_mask = to_pil_image(GSAM_mask)
                save_mask.save(f"{os.path.join(mask_save_path, file).replace('.jpg', '_mask.jpg')}")
                

if __name__ == "__main__":
    
    conf_path = sys.argv[1]
    conf = OmegaConf.load(conf_path)
    main(conf)

View source record →