Port-model / inference_example.py
Sunix2026's picture
Upload YOLOv11 segmentation model with config
b452cdf verified
raw
history blame contribute delete
632 Bytes
# Sample inference script
from ultralytics import YOLO
import json
# Load configuration
with open('config.json', 'r') as f:
config = json.load(f)
# Load model
model = YOLO('model.pt')
# Run inference with config parameters
results = model(
'your_image.jpg',
conf=config['confidence_threshold'],
iou=config['iou_threshold'],
imgsz=config['input_size']
)
# Process results
for result in results:
for box in result.boxes:
class_id = int(box.cls)
class_name = config['id2label'][str(class_id)]
confidence = float(box.conf)
print(f"Detected: {class_name} ({confidence:.2%})")