mindee
/

fasterrcnn_mobilenet_v3_large_fpn

Object Detection

Model card Files Files and versions Community

fg-mindee commited on Mar 7, 2022

Commit

b554400

•

1 Parent(s): 6906749

docs: Updated README

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -48,9 +48,9 @@ pip install -e doctr/.[torch]
 from PIL import Image
 import torch
 from torchvision.transforms import Compose, ConvertImageDtype, PILToTensor
-from holocron.models import model_from_hf_hub
-model = model_from_hf_hub("mindee/fasterrcnn_mobilenet_v3_large_fpn").eval()
 img = Image.open(path_to_an_image).convert("RGB")

 from PIL import Image
 import torch
 from torchvision.transforms import Compose, ConvertImageDtype, PILToTensor
+from doctr.models.obj_detection.factory import from_hub
+model = from_hub("mindee/fasterrcnn_mobilenet_v3_large_fpn").eval()
 img = Image.open(path_to_an_image).convert("RGB")