Add benchmark script for segment anything v2 #22169
Conversation
![github-advanced-security[bot]](https://avatars.githubusercontent.com/in/57789?s=60&v=4){kind=small}
| print(f"Start {config.repeats} runs of performance tests...") | ||
| start = time.time() | ||
| for _ in range(config.repeats): | ||
| _image_features_0, _image_features_1, _image_embeddings = sam2_encoder(img) |
Check notice
Code scanning / CodeQL
Unused local variable Note
| print(f"Start {config.repeats} runs of performance tests...") | ||
| start = time.time() | ||
| for _ in range(config.repeats): | ||
| _image_features_0, _image_features_1, _image_embeddings = sam2_encoder(img) |
Check notice
Code scanning / CodeQL
Unused local variable Note
| print(f"Start {config.repeats} runs of performance tests...") | ||
| start = time.time() | ||
| for _ in range(config.repeats): | ||
| _image_features_0, _image_features_1, _image_embeddings = sam2_encoder(img) |
Check notice
Code scanning / CodeQL
Unused local variable Note
| print(f"Start {config.repeats} runs of performance tests...") | ||
| start = time.time() | ||
| for _ in range(config.repeats): | ||
| _masks, _iou_predictions, _low_res_masks = sam2_decoder(*torch_inputs) |
Check notice
Code scanning / CodeQL
Unused local variable Note
| print(f"Start {config.repeats} runs of performance tests...") | ||
| start = time.time() | ||
| for _ in range(config.repeats): | ||
| _masks, _iou_predictions, _low_res_masks = sam2_decoder(*torch_inputs) |
Check notice
Code scanning / CodeQL
Unused local variable Note
| print(f"Start {config.repeats} runs of performance tests...") | ||
| start = time.time() | ||
| for _ in range(config.repeats): | ||
| _masks, _iou_predictions, _low_res_masks = sam2_decoder(*torch_inputs) |
Check notice
Code scanning / CodeQL
Unused local variable Note
Description
Add benchmark script segment anything v2.
It depends on #22119 for onnx export, and #22167 for sam2 graph fusion.
Motivation and Context
Benchmark SAM2 model performance.