face-detector-deepface / pages /2_Video_Detection_Results.py
sergey21000's picture
Upload 11 files
545f568 verified
raw
history blame
995 Bytes
from pathlib import Path
import pandas as pd
import streamlit as st
from draw_utils import draw_plots
st.set_page_config(
layout='wide',
initial_sidebar_state='auto',
page_title='Face Emotion Recognition',
page_icon='👻',
)
st.write("#### Отображение результатов детекции видео")
csv_path = 'video_annotations.csv'
@st.cache_data
def load_df():
df = pd.read_csv(csv_path)
return df
if Path('video_annotations.csv').exists():
df = load_df()
st.download_button(
label='Скачать csv аннотации',
data=df.to_csv().encode('utf-8'),
file_name=csv_path,
mime='text/csv',
)
draw_plots(df)
else:
st.write(
'#### Для отображения результатов детекции выберите видео' \
'на вкладке Video Detection и нажмите "Детекция видео"'
)