Spaces:
Running
Running
Dan Biagini
commited on
Commit
·
38d30c3
1
Parent(s):
7efb926
polish on breeds and menu
Browse files- src/Hockey_Breeds.py +1 -1
- src/app.py +2 -2
- src/hockey_object_detection.py +15 -0
src/Hockey_Breeds.py
CHANGED
@@ -57,7 +57,7 @@ st.image("src/images/artifacts/confusion_matrix.png", caption="Confusion Matrix
|
|
57 |
|
58 |
st.subheader("Try It Out")
|
59 |
|
60 |
-
img = image_select(label="Select an image and hockey breeds will guess a label", images=list(st.session_state.sample.values()))
|
61 |
|
62 |
st.button("Re-roll Samples", on_click=reroll_sample_images)
|
63 |
|
|
|
57 |
|
58 |
st.subheader("Try It Out")
|
59 |
|
60 |
+
img = image_select(label="Select an image and hockey breeds will guess a label. See if you can find some incorrect guesses...", images=list(st.session_state.sample.values()))
|
61 |
|
62 |
st.button("Re-roll Samples", on_click=reroll_sample_images)
|
63 |
|
src/app.py
CHANGED
@@ -5,8 +5,8 @@ import logging
|
|
5 |
app = st.navigation(
|
6 |
{"App": [st.Page("Home.py", title="Home", icon=":material/home:"),
|
7 |
st.Page("About.py", icon="🥅")],
|
8 |
-
"Models": [st.Page("Hockey_Breeds.py", icon=":material/gradient:"),
|
9 |
-
st.Page("hockey_object_detection.py", title="Hockey Object Detection", icon=":material/filter_b_and_w:")]
|
10 |
}
|
11 |
)
|
12 |
app.run()
|
|
|
5 |
app = st.navigation(
|
6 |
{"App": [st.Page("Home.py", title="Home", icon=":material/home:"),
|
7 |
st.Page("About.py", icon="🥅")],
|
8 |
+
"Models": [st.Page("Hockey_Breeds.py", title="v1 - Hockey Breeds", icon=":material/gradient:"),
|
9 |
+
st.Page("hockey_object_detection.py", title="v2 - Hockey Object Detection", icon=":material/filter_b_and_w:")]
|
10 |
}
|
11 |
)
|
12 |
app.run()
|
src/hockey_object_detection.py
CHANGED
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
|
3 |
+
st.set_page_config(page_title='Hockey Breeds v2 - Objects', layout="wide",
|
4 |
+
page_icon=":frame_with_picture:")
|
5 |
+
|
6 |
+
st.title('Hockey Breeds v2 - Objects')
|
7 |
+
st.markdown('The first version of Hockey Breeds was fun and educational, but not useful for analyzing hockey videos. The second version needs to be able to recognize individual "objects" within an image and ultimately track those objects through game play')
|
8 |
+
st.subheader('Object Detection')
|
9 |
+
|
10 |
+
desc = '''Hockey Breed detector v2 uses a state of the art (circa 2023) computer vision approach.
|
11 |
+
|
12 |
+
I used the same training images as the first version of the Hockey Breeds model, but change the ML algorithm to use YOLO object detection (YOLO v8).
|
13 |
+
The output will be a set of hockey objects (defined by "bounding boxes") with labels for any hockey image uploaded.'''
|
14 |
+
|
15 |
+
st.markdown(desc)
|