No such file or directory: './input/sam_vit_h_4b8939.pth'

boingboing·2023년 4월 12일
0

from segment_anything import SamPredictor, sam_model_registry


folder_path = '/content/drive/MyDrive/Colab Notebooks/pathology_ai/input/'

# model type -> default / vit_h / vit_l / vit_b 


# load SAM model 
sam_checkpoint = folder_path + "sam_vit_h_4b8939.pth" 
model_type = "vit_h" 
device = "cuda"

sam = sam_model_registry[model_type](checkpoint=sam_checkpoint) 
# checkpoint -> save the state of your model. These checkpoints contain the learned weights and biases of the model, which are updated during the training process.

에러 발생!!!!!

상대 경로로는 일단 안 잡힘.

절대경로로 잡으려는데, colab에서 돌리니까

Colab Notebook폴더가 아닌 바깥의, 구글 드라이브에 잡혀 있었음.

-> colab Notebook 폴더 안으로 옮김.

-> 그런데도 안됨

인터넷에서 wget으로 다운받은거 봄

해결

colab상에서도 wget으로 받아서 처리하니 해결됨. 그런데 드라이브상엔 안보이는데 어떻게 된걸까..?

어디 저장된거지..

0개의 댓글