stanfordmimi
/

synthpose-vitpose-base-hf

Inference Endpoints

Model card Files Files and versions Community

yonigozlan HF staff commited on 27 days ago

Commit

db32f8b

·

verified ·

1 Parent(s): 74722a9

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -128,8 +128,8 @@ person_boxes[:, 3] = person_boxes[:, 3] - person_boxes[:, 1]
 # Stage 2. Detect keypoints for each person found
 # ------------------------------------------------------------------------
-image_processor = AutoProcessor.from_pretrained("yonigozlan/synthpose-vit-base-hf")
-model = VitPoseForPoseEstimation.from_pretrained("yonigozlan/synthpose-vit-base-hf", device_map=device)
 inputs = image_processor(image, boxes=[person_boxes], return_tensors="pt").to(device)

 # Stage 2. Detect keypoints for each person found
 # ------------------------------------------------------------------------
+image_processor = AutoProcessor.from_pretrained("yonigozlan/synthpose-vitpose-base-hf")
+model = VitPoseForPoseEstimation.from_pretrained("yonigozlan/synthpose-vitpose-base-hf", device_map=device)
 inputs = image_processor(image, boxes=[person_boxes], return_tensors="pt").to(device)