(save_path, data)
| 171 | sava_numpy(save_path, curr) |
| 172 | |
| 173 | def sava_numpy(save_path, data): |
| 174 | save_path_s = save_path + '_s.npy' |
| 175 | save_path_f = save_path + '_f.npy' |
| 176 | np.save(save_path_s, data['seg_maps'].numpy()) |
| 177 | np.save(save_path_f, data['feature'].numpy()) |
| 178 | |
| 179 | def _embed_clip_sam_tiles(image, sam_encoder): |
| 180 | aug_imgs = torch.cat([image]) |