egeozsoy
/

EndoViT

Image Feature Extraction

Model card Files Files and versions Community

Ege Oezsoy commited on Apr 11

Commit

0cdd25f

•

1 Parent(s): 034b461

Adjustments

Files changed (2) hide show

config.json +0 -0
endovit_demo.py +1 -1

config.json ADDED Viewed

File without changes

endovit_demo.py CHANGED Viewed

@@ -45,7 +45,7 @@ images = torch.stack([process_single_image(image_path) for image_path in image_p
 device = "cuda"
 dtype = torch.float16
-model, loading_info = load_model_from_huggingface("egeozsoy/EndoViT", "endovit.pth")
 model = model.to(device, dtype)
 print(loading_info)
 output = model.forward_features(images.to(device, dtype))

 device = "cuda"
 dtype = torch.float16
+model, loading_info = load_model_from_huggingface("egeozsoy/EndoViT", "pytorch_model.bin")
 model = model.to(device, dtype)
 print(loading_info)
 output = model.forward_features(images.to(device, dtype))