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으로 받아서 처리하니 해결됨. 그런데 드라이브상엔 안보이는데 어떻게 된걸까..?
어디 저장된거지..