Skip to content

Commit

Permalink
update Hugging Face section in README.md to add_new_points_or_box
Browse files Browse the repository at this point in the history
  • Loading branch information
ronghanghu committed Aug 7, 2024
1 parent 2c21e4c commit 4afd912
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
state = predictor.init_state(<your_video>)

# add new prompts and instantly get the output on the same frame
frame_idx, object_ids, masks = predictor.add_new_points(state, <your_prompts>):
frame_idx, object_ids, masks = predictor.add_new_points_or_box(state, <your_prompts>):

# propagate the prompts to get masklets throughout the video
for frame_idx, object_ids, masks in predictor.propagate_in_video(state):
Expand Down

0 comments on commit 4afd912

Please sign in to comment.