rm -rf segmenter fold, only save checkpoints fold

This commit is contained in:
memoryunreal
2023-04-13 20:26:07 +00:00
parent a5606340a1
commit 5b808429c9

View File

@@ -45,7 +45,7 @@ class SamControler():
initialize sam controler
'''
checkpoint_url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
folder ="segmenter/checkpoints"
folder ="checkpoints"
SAM_checkpoint= 'sam_vit_h_4b8939.pth'
SAM_checkpoint = download_checkpoint(checkpoint_url, folder, SAM_checkpoint)
# SAM_checkpoint = '/ssd1/gaomingqi/checkpoints/sam_vit_h_4b8939.pth'