Views
No views yet
1# label_id: label_name
2{0: "AADHAR_NUMBER", 1: "DATE_OF_BIRTH", 2: "GENDER", 3: "NAME", 4: "ADDRESS"}$ pip install ultralytics huggingface_hub supervision1from ultralytics import YOLO
2from huggingface_hub import snapshot_download
3from supervision import Detections
4
5MODEL_REPO_ID = "thewalnutaisg/YOLOv8-aadhar-card-int8-openvino"
6MODEL_LOCAL_DIR = "./models/model_int8_openvino_model"
7
8snapshot_download(
9 repo_id=MODEL_REPO_ID,
10 local_dir=MODEL_LOCAL_DIR,
11)
12
13image_url = "./Sample 1/100001005885477 (1).jpg"
14
15model = YOLO(MODEL_LOCAL_DIR, task="detect")
16detections = Detections.from_ultralytics(model.predict(image_url)[0])
17
18print(detections)