add YOLOv10_project

dohuyduc2002 · Jun 27, 2024 · b9f7ffd · b9f7ffd
1 parent 23c678f
commit b9f7ffd
Show file tree

Hide file tree

Showing 34 changed files with 50 additions and 1,373 deletions.
diff --git a/Week_1/activate_function.py → Module_1/Week_1/activate_function.py b/Week_1/activate_function.py → Module_1/Week_1/activate_function.py
diff --git a/Week_1/calculate_f1_score.py → Module_1/Week_1/calculate_f1_score.py b/Week_1/calculate_f1_score.py → Module_1/Week_1/calculate_f1_score.py
diff --git a/Week_1/hyperbolic_approx.py → Module_1/Week_1/hyperbolic_approx.py b/Week_1/hyperbolic_approx.py → Module_1/Week_1/hyperbolic_approx.py
diff --git a/Week_1/md_rne.py → Module_1/Week_1/md_rne.py b/Week_1/md_rne.py → Module_1/Week_1/md_rne.py
diff --git a/Week_1/multiple_choice_week1.ipynb → Module_1/Week_1/multiple_choice_week1.ipynb b/Week_1/multiple_choice_week1.ipynb → Module_1/Week_1/multiple_choice_week1.ipynb
diff --git a/Week_1/regression_loss.py → Module_1/Week_1/regression_loss.py b/Week_1/regression_loss.py → Module_1/Week_1/regression_loss.py
diff --git a/Week_1/test_case_week1.py → Module_1/Week_1/test_case_week1.py b/Week_1/test_case_week1.py → Module_1/Week_1/test_case_week1.py
diff --git a/...2/__pycache__/count_chars.cpython-311.pyc → ...2/__pycache__/count_chars.cpython-311.pyc b/...2/__pycache__/count_chars.cpython-311.pyc → ...2/__pycache__/count_chars.cpython-311.pyc
diff --git a/...2/__pycache__/levenshtein.cpython-311.pyc → ...2/__pycache__/levenshtein.cpython-311.pyc b/...2/__pycache__/levenshtein.cpython-311.pyc → ...2/__pycache__/levenshtein.cpython-311.pyc
diff --git a/..._pycache__/sliding_window.cpython-311.pyc → ..._pycache__/sliding_window.cpython-311.pyc b/..._pycache__/sliding_window.cpython-311.pyc → ..._pycache__/sliding_window.cpython-311.pyc
diff --git a/..._2/__pycache__/word_count.cpython-311.pyc → ..._2/__pycache__/word_count.cpython-311.pyc b/..._2/__pycache__/word_count.cpython-311.pyc → ..._2/__pycache__/word_count.cpython-311.pyc
diff --git a/Week_3/__pycache__/my_queue.cpython-39.pyc → ...eek_3/__pycache__/my_queue.cpython-39.pyc b/Week_3/__pycache__/my_queue.cpython-39.pyc → ...eek_3/__pycache__/my_queue.cpython-39.pyc
diff --git a/Week_3/__pycache__/my_stack.cpython-39.pyc → ...eek_3/__pycache__/my_stack.cpython-39.pyc b/Week_3/__pycache__/my_stack.cpython-39.pyc → ...eek_3/__pycache__/my_stack.cpython-39.pyc
diff --git a/Week_3/__pycache__/queue.cpython-311.pyc → .../Week_3/__pycache__/queue.cpython-311.pyc b/Week_3/__pycache__/queue.cpython-311.pyc → .../Week_3/__pycache__/queue.cpython-311.pyc
diff --git a/Week_3/__pycache__/queue.cpython-39.pyc → ...1/Week_3/__pycache__/queue.cpython-39.pyc b/Week_3/__pycache__/queue.cpython-39.pyc → ...1/Week_3/__pycache__/queue.cpython-39.pyc
diff --git a/.../__pycache__/torch_softmax.cpython-39.pyc → .../__pycache__/torch_softmax.cpython-39.pyc b/.../__pycache__/torch_softmax.cpython-39.pyc → .../__pycache__/torch_softmax.cpython-39.pyc
diff --git a/Week_3/__pycache__/ward_oop.cpython-39.pyc → ...eek_3/__pycache__/ward_oop.cpython-39.pyc b/Week_3/__pycache__/ward_oop.cpython-39.pyc → ...eek_3/__pycache__/ward_oop.cpython-39.pyc
diff --git a/Module_1/YOLOv10_project/deploy.py b/Module_1/YOLOv10_project/deploy.py
@@ -0,0 +1,31 @@
+import streamlit as st
+import cv2
+import numpy as np
+from PIL import Image
+import streamlit as st
+from ultralytics import YOLOv10
+
+MODEL_PATH = "./models/best.pt"
+
+
+def process_image(image):
+    model = YOLOv10(MODEL_PATH)
+    result = model(image)[0]
+    result.save('./predicts/output.png')
+
+
+def main():
+    st.title('Object Detection for Images')
+    file = st.file_uploader('Upload Image', type=['jpg', 'png', 'jpeg'])
+    if file is not None:
+        st.image(file, caption="Uploaded Image")
+
+        image = Image.open(file)
+        image = np.array(image)
+        process_image(image)
+        predict_image = cv2.imread('./predicts/output.png')
+        st.image(predict_image, caption="Predict Image")
+
+
+if __name__ == "__main__":
+    main()
diff --git a/Module_1/YOLOv10_project/models/best.pt b/Module_1/YOLOv10_project/models/best.pt
diff --git a/Module_1/YOLOv10_project/models/yolo-v10.ipynb b/Module_1/YOLOv10_project/models/yolo-v10.ipynb
diff --git a/Module_1/YOLOv10_project/project_requirements.txt b/Module_1/YOLOv10_project/project_requirements.txt
@@ -0,0 +1,17 @@
+torch==2.0.1
+torchvision==0.15.2
+onnx==1.14.0
+onnxruntime==1.15.1
+pycocotools==2.0.7
+PyYAML==6.0.1
+scipy==1.13.0
+onnxsim==0.4.36
+onnxruntime-gpu==1.18.0
+gradio==4.31.5
+opencv-python==4.9.0.80
+psutil==5.9.8
+py-cpuinfo==9.0.0
+huggingface-hub==0.23.2
+safetensors==0.4.3
+streamlit
+ultralytics
diff --git a/Module_1/YOLOv10_project/yolov10 b/Module_1/YOLOv10_project/yolov10
diff --git a/Week_2/P1_data.txt b/Week_2/P1_data.txt
diff --git a/Week_2/count_chars.py b/Week_2/count_chars.py
diff --git a/Week_2/levenshtein.py b/Week_2/levenshtein.py