fcakyon commited on
Commit
486e870
β€’
1 Parent(s): f995a01
Files changed (2) hide show
  1. app.py +23 -5
  2. utils.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import sahi.utils.mmdet
3
  import sahi.model
4
  from PIL import Image
5
  import random
6
- from utils import image_compare
7
  from utils import sahi_mmdet_inference
8
 
9
  MMDET_YOLACT_MODEL_URL = "https://download.openmmlab.com/mmdetection/v2.0/yolact/yolact_r50_1x8_coco/yolact_r50_1x8_coco_20200908-f38d58df.pth"
@@ -121,7 +121,7 @@ st.markdown(
121
  <p style='text-align: center'>
122
  <a href='https://github.com/obss/sahi' target='_blank'>SAHI Github</a> | <a href='https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox' target='_blank'>YOLOX Github</a> | <a href='https://huggingface.co/spaces/fcakyon/sahi-yolov5' target='_blank'>SAHI+YOLOv5 Demo</a>
123
  <br />
124
- Follow me for more! <a href='https://twitter.com/fcakyon' target='_blank'>twitter</a> | <a href='https://www.linkedin.com/in/fcakyon/' target='_blank'>linkedin</a> | <a href='https://fcakyon.medium.com/' target='_blank'>medium</a>
125
  </p>
126
  """,
127
  unsafe_allow_html=True,
@@ -129,6 +129,24 @@ st.markdown(
129
 
130
  st.write("##")
131
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
132
  col1, col2, col3 = st.columns([6, 1, 6])
133
  with col1:
134
  st.markdown(f"##### Set input image:")
@@ -181,9 +199,9 @@ with col3:
181
  )
182
  postprocess_class_agnostic = st.checkbox("postprocess_class_agnostic", value=True)
183
 
184
- col1, col2, col3 = st.columns([6, 1, 6])
185
  with col2:
186
- submit = st.button("Submit")
187
 
188
  if submit:
189
  # perform prediction
@@ -219,7 +237,7 @@ if submit:
219
  st.session_state["output_2"] = output_2
220
 
221
  st.markdown(f"##### YOLOX Standard vs SAHI Prediction:")
222
- static_component = image_compare(
223
  img1=st.session_state["output_1"],
224
  img2=st.session_state["output_2"],
225
  label1="YOLOX",
3
  import sahi.model
4
  from PIL import Image
5
  import random
6
+ from utils import image_comparison
7
  from utils import sahi_mmdet_inference
8
 
9
  MMDET_YOLACT_MODEL_URL = "https://download.openmmlab.com/mmdetection/v2.0/yolact/yolact_r50_1x8_coco/yolact_r50_1x8_coco_20200908-f38d58df.pth"
121
  <p style='text-align: center'>
122
  <a href='https://github.com/obss/sahi' target='_blank'>SAHI Github</a> | <a href='https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox' target='_blank'>YOLOX Github</a> | <a href='https://huggingface.co/spaces/fcakyon/sahi-yolov5' target='_blank'>SAHI+YOLOv5 Demo</a>
123
  <br />
124
+ Follow me for more! <a href='https://twitter.com/fcakyon' target='_blank'> <img src="https://img.icons8.com/color/48/000000/twitter--v1.png" height="30"></a><a href='https://www.linkedin.com/in/fcakyon/' target='_blank'><img src="https://img.icons8.com/fluency/48/000000/linkedin.png" height="30"></a> <a href='https://fcakyon.medium.com/' target='_blank'><img src="https://img.icons8.com/ios-filled/48/000000/medium-monogram.png" height="27"></a>
125
  </p>
126
  """,
127
  unsafe_allow_html=True,
129
 
130
  st.write("##")
131
 
132
+ with st.expander("Usage"):
133
+ st.markdown(
134
+ """
135
+ <p>
136
+ 1. Upload or select the input image πŸ–ΌοΈ
137
+ <br />
138
+ 2. (Optional) Set SAHI parameters βœ”οΈ
139
+ <br />
140
+ 3. Press to "πŸš€ Perform Prediction"
141
+ <br />
142
+ 4. Enjoy sliding image comparison πŸ”₯
143
+ </p>
144
+ """,
145
+ unsafe_allow_html=True,
146
+ )
147
+
148
+ st.write("##")
149
+
150
  col1, col2, col3 = st.columns([6, 1, 6])
151
  with col1:
152
  st.markdown(f"##### Set input image:")
199
  )
200
  postprocess_class_agnostic = st.checkbox("postprocess_class_agnostic", value=True)
201
 
202
+ col1, col2, col3 = st.columns([4, 3, 4])
203
  with col2:
204
+ submit = st.button("πŸš€ Perform Prediction")
205
 
206
  if submit:
207
  # perform prediction
237
  st.session_state["output_2"] = output_2
238
 
239
  st.markdown(f"##### YOLOX Standard vs SAHI Prediction:")
240
+ static_component = image_comparison(
241
  img1=st.session_state["output_1"],
242
  img2=st.session_state["output_2"],
243
  label1="YOLOX",
utils.py CHANGED
@@ -86,7 +86,7 @@ def pillow_local_file_to_base64(image: Image.Image):
86
  return base64_src
87
 
88
 
89
- def image_compare(
90
  img1: str,
91
  img2: str,
92
  label1: str = "1",
86
  return base64_src
87
 
88
 
89
+ def image_comparison(
90
  img1: str,
91
  img2: str,
92
  label1: str = "1",