Upload 4 files

app.py

@@ -0,0 +1,71 @@
+import streamlit as st
+import subprocess
+import os
+import time
+import pandas as pd
+from PIL import Image
+
+def run_command(command):
+    process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
+    process.communicate()
+
+def main():
+    st.set_page_config(
+        page_title="OCR Captcha Solver",
+        page_icon=":books:",
+        layout="wide",
+        initial_sidebar_state="expanded",
+    )
+
+    # Add title in the middle of the screen
+    st.write(
+        """
+        # OCR Captcha Solver
+        """
+    )
+    
+    st.sidebar.title("Options")
+    st.sidebar.markdown("Upload an image and click the button to run text recognition.")
+    
+    uploaded_file = st.sidebar.file_uploader("Upload an image", type=["jpg", "png"])
+
+    if uploaded_file is not None:
+        image_folder = "./user_images"
+        os.makedirs(image_folder, exist_ok=True)
+
+        image_path = os.path.join(image_folder, uploaded_file.name)
+        with open(image_path, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        st.success(f"Image successfully saved to {image_path}")
+
+        command = f"python C:/Users/Dhanush/Downloads/staret/deep-text-recognition-benchmark/demo.py > C:/Users/Dhanush/Downloads/staret/output.txt --image_folder C:/Users/Dhanush/Downloads/staret/user_images --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --saved_model C:/Users/Dhanush/Downloads/staret/best_accuracy.pth"
+
+        if st.sidebar.button("Run Text Recognition"):
+            with st.spinner("Running... Please wait."):
+                run_command(command)
+                time.sleep(3)
+
+        output_path = "C:/Users/Dhanush/Downloads/staret/output.txt"
+        if os.path.isfile(output_path):
+            with open(output_path, "r") as file:
+                output_lines = file.readlines()
+
+            if len(output_lines) > 5:
+                data = []
+                for line in output_lines[5:]:
+                    line_data = line.strip().split('\t')
+                    data.append(line_data)
+                
+                df = pd.DataFrame(data, columns=["Image Path", "Predicted Labels", "Confidence Score"])
+                st.table(df)
+                
+                # Display uploaded image
+                uploaded_image = Image.open(uploaded_file)
+                st.image(uploaded_image, caption="Uploaded Image", use_column_width=True)
+            else:
+                st.write("No results found.")
+        else:
+            st.write("The process is still running. Please wait for a moment and refresh the page.")
+
+if __name__ == "__main__":
+    main()

best_accuracy.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a12309e2c9cb901cae996744f692297f5be3871aa8d6baeedcebf237a9759ff
+size 198717893

log_demo_result.txt

@@ -0,0 +1,56 @@
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/data/sample\2b827.png	2b827                    	0.9994
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/data/sample\2b827.png	2b827                    	0.9994
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/data/sample\2b827.png	2b827                    	0.9994
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+C:/Users/Dhanush/Downloads/staret/user_images\2g7nm.png	2g7nm                    	0.9994
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+C:/Users/Dhanush/Downloads/staret/user_images\2g7nm.png	2g7nm                    	0.9994
+C:/Users/Dhanush/Downloads/staret/user_images\download.png	xff8d                    	0.0354
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+C:/Users/Dhanush/Downloads/staret/user_images\2g7nm.png	2g7nm                    	0.9994
+C:/Users/Dhanush/Downloads/staret/user_images\6m5eg.png	6m5eg                    	0.9999
+C:/Users/Dhanush/Downloads/staret/user_images\download.png	xff8d                    	0.0354
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+C:/Users/Dhanush/Downloads/staret/user_images\2g7nm.png	2g7nm                    	0.9994
+C:/Users/Dhanush/Downloads/staret/user_images\6m5eg.png	6m5eg                    	0.9999
+C:/Users/Dhanush/Downloads/staret/user_images\74853.png	74853                    	0.9999
+C:/Users/Dhanush/Downloads/staret/user_images\download.png	xff8d                    	0.0354
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+C:/Users/Dhanush/Downloads/staret/user_images\2g7nm.png	2g7nm                    	0.9994
+C:/Users/Dhanush/Downloads/staret/user_images\6m5eg.png	6m5eg                    	0.9999
+C:/Users/Dhanush/Downloads/staret/user_images\8n34n.png	8n34n                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\74853.png	74853                    	0.9999

output.txt

@@ -0,0 +1,11 @@
+model input parameters 32 100 20 1 512 256 38 25 TPS ResNet BiLSTM Attn
+loading pretrained model from C:/Users/Dhanush/Downloads/staret/best_accuracy.pth
+--------------------------------------------------------------------------------
+image_path               	predicted_labels         	confidence score
+--------------------------------------------------------------------------------
+C:/Users/Dhanush/Downloads/staret/user_images\2enf4.png	2enf4                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\2fxgd.png	2fxgd                    	0.9998
+C:/Users/Dhanush/Downloads/staret/user_images\2g7nm.png	2g7nm                    	0.9994
+C:/Users/Dhanush/Downloads/staret/user_images\6m5eg.png	6m5eg                    	0.9999
+C:/Users/Dhanush/Downloads/staret/user_images\8n34n.png	8n34n                    	1.0000
+C:/Users/Dhanush/Downloads/staret/user_images\74853.png	74853                    	0.9999