File size: 2,112 Bytes
6a3a328 14a2f01 6a3a328 5da8fea b61deaf a3f3e32 6a3a328 a3f3e32 6a3a328 a3f3e32 6a3a328 d0dd251 6a3a328 0f897c1 d0dd251 6a3a328 d0dd251 6a3a328 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
-------------------------------------------------
@File Name: app.py
@Author: Luyao.zhang
@Date: 2023/5/15
@Description:
-------------------------------------------------
"""
from pathlib import Path
from PIL import Image
import streamlit as st
import config
from utils import load_model, infer_uploaded_image, infer_uploaded_video, infer_uploaded_webcam
from download_gd import *
import os
if not os.path.exists('./crowded_human_yolov8s.pt'):
download_file_from_google_drive('1qCXBDy3YuxS9bqfRIc--cRLo1L3DAYq2', './crowded_human_yolov8s.pt')
if os.path.exists('./crowded_human_yolov8s.pt'):
file_exist = "./crowded_human_yolov8s.pt"
else:
file_exist = "No file"
# setting page layout
st.set_page_config(
page_title="YOLO.dog",
page_icon="🤖",
layout="wide",
initial_sidebar_state="expanded"
)
# main page heading
st.title(file_exist)
# sidebar
st.sidebar.header("DL Model Config")
# model options
task_type = "Detection"
model_type = "crowded_human"
confidence = float(st.sidebar.slider(
"Select Model Confidence", 30, 100, 50)) / 100
if model_type:
#model_path = Path(config.DETECTION_MODEL_DIR, str(model_type))
model_path = "./crowded_human_yolov8s.pt"
else:
st.error("Please Select Model in Sidebar")
# load pretrained DL model
try:
print('model_path', model_path)
model = load_model(model_path)
except Exception as e:
st.error(f"Unable to load model. Please check the specified path: {model_path}")
# image/video options
st.sidebar.header("Image/Video Config")
source_selectbox = st.sidebar.selectbox(
"Select Source",
config.SOURCES_LIST
)
source_img = None
if source_selectbox == config.SOURCES_LIST[0]: # Image
infer_uploaded_image(confidence, model)
elif source_selectbox == config.SOURCES_LIST[1]: # Video
infer_uploaded_video(confidence, model)
elif source_selectbox == config.SOURCES_LIST[2]: # Webcam
infer_uploaded_webcam(confidence, model)
else:
st.error("Currently only 'Image' and 'Video' source are implemented")
|