File size: 1,106 Bytes
b9e956e 1fb4354 b9e956e 1fb4354 b9e956e 1fb4354 |
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 |
import streamlit as st
if __name__ == "__main__":
# set page configurations and display/annotation options
st.set_page_config(
page_title="Circuit Sketch Recognizer",
layout="wide"
)
st.title("Circuit Sketch Recognition")
st.markdown(
'''
Can computers recognize hand-drawn circuit sketches? Absolutely!
Upload a picture of your circuit sketch or take a new one to see it in action.
This project leverages fine tuned TrOCR for text recognition and YOLOv8 for component detection, using CGHD-2304 dataset.
'''
)
col1, col2 = st.columns(2)
with col1:
st.image('media/capture.gif', use_column_width=True, caption='Take a Picture')
with col2:
st.image('media/upload.gif', use_column_width=True, caption='Upload an Image')
st.markdown('Here are some more examples!')
col3, col4 = st.columns(2)
with col3:
st.image('media/example1.jpg', use_column_width=True, caption='Example 1')
with col4:
st.image('media/example2.jpg', use_column_width=True, caption='Example 2')
|