Beasto/HumanSegmentation

1from PIL import Image,ImageDraw,ImageFont2import numpy as np3from ultralytics import YOLO4import streamlit as st5 6model = YOLO('HumanSegmentation.pt')7img = st.file_uploader('Choose an Image')8 9if img is not None:10    img = Image.open(img).resize((640,640)).convert('RGB')11    results = model(img)12    st.image(img,'Input')13    for result in results:14        for mask in result.masks.data:15            mask = mask.numpy()16            st.image(mask,'Mask')17