Spaces:
Sleeping
Sleeping
| from PIL import Image,ImageDraw,ImageFont | |
| import numpy as np | |
| from ultralytics import YOLO | |
| import streamlit as st | |
| model = YOLO('VehicleDetection.pt') | |
| img = st.file_uploader('Choose an Image') | |
| if img is not None: | |
| img = Image.open(img).resize((640,640)).convert('RGB') | |
| results = model(img) | |
| for result in results: | |
| for i in range(len(result.boxes.cls)): | |
| boxes = result.boxes.xyxy[i] | |
| draw = ImageDraw.Draw(img) | |
| draw.rectangle([boxes[0], boxes[1], boxes[2], boxes[3]], outline="black", width=5) | |
| st.image(img,'Detected') |