diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..dedff2f70f5f07ab34218deb4f0059ecaffc4fe2 100644 --- a/.gitattributes +++ b/.gitattributes @@ -33,3 +33,48 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text *.zip filter=lfs diff=lfs merge=lfs -text *.zst filter=lfs diff=lfs merge=lfs -text *tfevents* filter=lfs diff=lfs merge=lfs -text +examples/0.jpg filter=lfs diff=lfs merge=lfs -text +examples/10.jpg filter=lfs diff=lfs merge=lfs -text +examples/11.jpg filter=lfs diff=lfs merge=lfs -text +examples/12.jpg filter=lfs diff=lfs merge=lfs -text +examples/13.jpg filter=lfs diff=lfs merge=lfs -text +examples/14.jpg filter=lfs diff=lfs merge=lfs -text +examples/15_SxfgB6I.jpg filter=lfs diff=lfs merge=lfs -text +examples/16_gAlDyfj.jpg filter=lfs diff=lfs merge=lfs -text +examples/16.jpg filter=lfs diff=lfs merge=lfs -text +examples/17_NdIwCTa.jpg filter=lfs diff=lfs merge=lfs -text +examples/18_2LN8Wae.jpg filter=lfs diff=lfs merge=lfs -text +examples/18.jpg filter=lfs diff=lfs merge=lfs -text +examples/19_bQOd8mo.jpg filter=lfs diff=lfs merge=lfs -text +examples/19.jpg filter=lfs diff=lfs merge=lfs -text +examples/2.jpg filter=lfs diff=lfs merge=lfs -text +examples/20_d4qEaSG.jpg filter=lfs diff=lfs merge=lfs -text +examples/20.jpg filter=lfs diff=lfs merge=lfs -text +examples/21_E6oxZY3.jpg filter=lfs diff=lfs merge=lfs -text +examples/21.jpg filter=lfs diff=lfs merge=lfs -text +examples/22_eimzU9L.jpg filter=lfs diff=lfs merge=lfs -text +examples/22.jpg filter=lfs diff=lfs merge=lfs -text +examples/23_XH0MPmx.jpg filter=lfs diff=lfs merge=lfs -text +examples/23.jpg filter=lfs diff=lfs merge=lfs -text +examples/24_6gMkQPj.jpg filter=lfs diff=lfs merge=lfs -text +examples/24.jpg filter=lfs diff=lfs merge=lfs -text +examples/25_owD4RvO.jpg filter=lfs diff=lfs merge=lfs -text +examples/25.jpg filter=lfs diff=lfs merge=lfs -text +examples/26.jpg filter=lfs diff=lfs merge=lfs -text +examples/27_X2pkXLB.jpg filter=lfs diff=lfs merge=lfs -text +examples/27.jpg filter=lfs diff=lfs merge=lfs -text +examples/28_FP3TIDL.jpg filter=lfs diff=lfs merge=lfs -text +examples/28.jpg filter=lfs diff=lfs merge=lfs -text +examples/29.jpg filter=lfs diff=lfs merge=lfs -text +examples/3.jpg filter=lfs diff=lfs merge=lfs -text +examples/30.jpg filter=lfs diff=lfs merge=lfs -text +examples/4.jpg filter=lfs diff=lfs merge=lfs -text +examples/5.jpg filter=lfs diff=lfs merge=lfs -text +examples/6.jpg filter=lfs diff=lfs merge=lfs -text +examples/8.jpg filter=lfs diff=lfs merge=lfs -text +examples/9.jpg filter=lfs diff=lfs merge=lfs -text +metrics/BoxF1_curve.png filter=lfs diff=lfs merge=lfs -text +metrics/BoxP_curve.png filter=lfs diff=lfs merge=lfs -text +metrics/BoxPR_curve.png filter=lfs diff=lfs merge=lfs -text +metrics/BoxR_curve.png filter=lfs diff=lfs merge=lfs -text +metrics/results.png filter=lfs diff=lfs merge=lfs -text diff --git a/README.md b/README.md index 9ed2d23c9862c90284c17000b535cdef1d266dee..6a3e80f48fb4d5777d6dae5876af310ec8c70d45 100644 --- a/README.md +++ b/README.md @@ -1,15 +1,43 @@ ---- -title: Warehouse Vision -emoji: ๐Ÿ“Š -colorFrom: blue -colorTo: gray -sdk: gradio -sdk_version: 6.19.0 -python_version: '3.12' -app_file: app.py -pinned: false -license: mit -short_description: AI-powered Warehouse Object Detection ---- - -Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference +--- +title: Warehouse Vision +emoji: ๐Ÿ“Š +colorFrom: blue +colorTo: gray +sdk: gradio +sdk_version: 6.19.0 +python_version: '3.12' +app_file: app.py +pinned: false +license: mit +short_description: AI-powered Warehouse Object Detection +--- +# Warehouse Vision AI + +A YOLOv8-based warehouse object detector trained on industrial warehouse scenes. + +## Features + +- Upload your own warehouse image +- Try 30 built-in sample images +- View model performance metrics +- Interactive confidence and IoU sliders + +## Model + +YOLOv8 Custom Trained + +## Dataset + +This project was trained using the SORDI Industrial Scene Dataset. + +Dataset source: +https://www.kaggle.com/datasets/sordi-ai/industrial-scene-04-dataset + +https://www.kaggle.com/datasets/sordi-ai/industrial-scene-03-dataset + +The datasets are provided under the MIT License. + +## Deployment + +Hugging Face Spaces + Gradio +Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference diff --git a/app.py b/app.py new file mode 100644 index 0000000000000000000000000000000000000000..a83889d5ea0b059012046cb703b5a2da40a26ab8 --- /dev/null +++ b/app.py @@ -0,0 +1,216 @@ +import os +from collections import Counter + +import cv2 +import gradio as gr +import numpy as np +from PIL import Image +from ultralytics import YOLO + +# ========================================================== +# Load Model +# ========================================================== + +model = YOLO("best.pt") + +# ========================================================== +# Example Images +# ========================================================== + +example_images = [] + +if os.path.exists("examples"): + for file in sorted(os.listdir("examples")): + if file.lower().endswith((".jpg", ".jpeg", ".png")): + example_images.append([os.path.join("examples", file)]) + +# ========================================================== +# Detection Function +# ========================================================== + +def detect(image, conf, iou): + + results = model.predict( + source=image, + conf=conf, + iou=iou, + verbose=False + ) + + result = results[0] + + plotted = result.plot() + + plotted = cv2.cvtColor(plotted, cv2.COLOR_BGR2RGB) + + detected = [] + + for cls in result.boxes.cls.tolist(): + detected.append(model.names[int(cls)]) + + counter = Counter(detected) + + table = [] + + for name, count in sorted(counter.items()): + table.append([name, count]) + + return Image.fromarray(plotted), table + +# ========================================================== +# Metric Images +# ========================================================== + +metric_files = [ + "metrics/results.png", + "metrics/P_curve.png", + "metrics/R_curve.png", + "metrics/PR_curve.png", + "metrics/F1_curve.png", + "metrics/confusion_matrix.png" +] + +metric_components = [] + +for path in metric_files: + if os.path.exists(path): + metric_components.append(gr.Image(value=path, label=os.path.basename(path))) + +# ========================================================== +# About Text +# ========================================================== + +about = """ +# Warehouse Vision AI + +### Industrial Object Detection using YOLOv8 + +This project detects warehouse objects using a custom-trained YOLO model. + +### Features + +- Industrial Rack Detection +- KLT Box Detection +- Worker Detection +- Safety Equipment Detection +- Warehouse Object Localization + +### Framework + +- Ultralytics YOLO +- PyTorch +- Gradio +- Hugging Face Spaces + +### Author + +Omkar Kalburgi +""" + +# ========================================================== +# UI +# ========================================================== + +with gr.Blocks(title="Warehouse Vision AI") as demo: + + gr.Markdown( + """ +# ๐Ÿ“ฆ Warehouse Vision AI + +### YOLO-based Industrial Warehouse Object Detection + +Upload an image or try one of the sample images. +""" + ) + + with gr.Tabs(): + + # -------------------------------------------------- + + with gr.Tab("๐Ÿ” Detection"): + + with gr.Row(): + + with gr.Column(): + + image = gr.Image(type="pil", label="Input Image") + + conf = gr.Slider( + 0.1, + 1.0, + value=0.25, + step=0.05, + label="Confidence Threshold", + ) + + iou = gr.Slider( + 0.1, + 1.0, + value=0.45, + step=0.05, + label="IoU Threshold", + ) + + btn = gr.Button("Run Detection") + + with gr.Column(): + + output = gr.Image(label="Prediction") + + table = gr.Dataframe( + headers=["Class", "Count"], + datatype=["str", "number"], + interactive=False, + label="Detected Objects", + ) + + btn.click( + detect, + inputs=[image, conf, iou], + outputs=[output, table], + ) + + # -------------------------------------------------- + + with gr.Tab("๐Ÿงช Sample Images"): + + gr.Markdown("Click any image below to test the model.") + + sample_input = gr.Image(type="pil") + + sample_output = gr.Image() + + sample_table = gr.Dataframe( + headers=["Class", "Count"], + interactive=False, + ) + + gr.Examples( + examples=example_images, + inputs=sample_input, + ) + + sample_btn = gr.Button("Run Detection") + + sample_btn.click( + detect, + inputs=[sample_input, conf, iou], + outputs=[sample_output, sample_table], + ) + + # -------------------------------------------------- + + with gr.Tab("๐Ÿ“ˆ Model Performance"): + + gr.Markdown("Training Metrics") + + for img in metric_components: + img.render() + + # -------------------------------------------------- + + with gr.Tab("๐Ÿ“– About"): + + gr.Markdown(about) + +demo.launch() \ No newline at end of file diff --git a/best.pt b/best.pt new file mode 100644 index 0000000000000000000000000000000000000000..28b1f01359fc461109e9640bd9e3fc8f5978f98e --- /dev/null +++ b/best.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d7b996a81f31abcb01df498bfca47e44299085da33a9c94f428177665dc03be2 +size 22531882 diff --git a/examples/0.jpg b/examples/0.jpg new file mode 100644 index 0000000000000000000000000000000000000000..74e62fd0f6afcda471176903aad9f0a58a7b20dd --- /dev/null +++ b/examples/0.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6940952f6f3b2ec317615a7444a1f01c6516624756fecb44a90d6e21175449bd +size 298747 diff --git a/examples/10.jpg b/examples/10.jpg new file mode 100644 index 0000000000000000000000000000000000000000..e8b59f614e917ae2ac948a1787df52bdc1b7828e --- /dev/null +++ b/examples/10.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:626818ef38058b4a169d79cd81f28ab466f75650bb22331ee47103f76bd0c1f7 +size 196342 diff --git a/examples/11.jpg b/examples/11.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d412bcdac4e96f19ba30aeecf930f26215775582 --- /dev/null +++ b/examples/11.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dceb9ec36afdd521f2c45785312c368df640f791933934c13a8d25c943539be8 +size 220792 diff --git a/examples/12.jpg b/examples/12.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5c5884857e401c6ed81209b3204fe10b9fd9cbe7 --- /dev/null +++ b/examples/12.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3c236948538fb8d34dc1492dbb2f7bd0e156bce4fbc8757a75dc8f27eba95821 +size 319047 diff --git a/examples/13.jpg b/examples/13.jpg new file mode 100644 index 0000000000000000000000000000000000000000..40e7f0a4b3235ca75d289e835df7453db1331f08 --- /dev/null +++ b/examples/13.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a7bd85b27dca52f971a8d477179e52ff636b2fbeef09d45f6d2ad715d1df57da +size 134199 diff --git a/examples/14.jpg b/examples/14.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ef2e02ef585d1230906d35ac9d45fb2d7c99edf7 --- /dev/null +++ b/examples/14.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8c8b002f38c0d7ebdec743e615bd0c8d8bd6b1ae7b0403ff784938f4050d48c +size 161088 diff --git a/examples/15.jpg b/examples/15.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3a29d64fed359d80f55d8127b477d2a53c0ca2c8 Binary files /dev/null and b/examples/15.jpg differ diff --git a/examples/15_SxfgB6I.jpg b/examples/15_SxfgB6I.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c9adb73eef2cc8005c378836829a8a2d57285ef7 --- /dev/null +++ b/examples/15_SxfgB6I.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a0ee9d73f4017eab39b1a47be3706eb9aee0de177c09e9b9407b9dc2dbd5767d +size 139408 diff --git a/examples/16.jpg b/examples/16.jpg new file mode 100644 index 0000000000000000000000000000000000000000..70adcf00bbfafed452766e939907b7a227923a97 --- /dev/null +++ b/examples/16.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01740327fcd8948cad0fdad8a305a5d102be791369c49123372d3564b2f8db9e +size 169412 diff --git a/examples/16_gAlDyfj.jpg b/examples/16_gAlDyfj.jpg new file mode 100644 index 0000000000000000000000000000000000000000..86dfd58646e4cc4f0dc4d292dca5204423bb7d3e --- /dev/null +++ b/examples/16_gAlDyfj.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d7c8a1caf8a825d70e3506e8e5a864ffacea22c3d4d7533c9d9e0fa8129fdd6e +size 217604 diff --git a/examples/17.jpg b/examples/17.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3f3ea3cdddfd1a96dd411dd273cf5d54da056208 Binary files /dev/null and b/examples/17.jpg differ diff --git a/examples/17_NdIwCTa.jpg b/examples/17_NdIwCTa.jpg new file mode 100644 index 0000000000000000000000000000000000000000..770989332709e6d2c86aa416d2add3fba46e6e9f --- /dev/null +++ b/examples/17_NdIwCTa.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:328797b8e60b0de9bbf8f651ed3893101e3a05712815f23202da3ab7987f8837 +size 185822 diff --git a/examples/18.jpg b/examples/18.jpg new file mode 100644 index 0000000000000000000000000000000000000000..55614b6c1eb3d023ebb5d283a827be2c9e030c16 --- /dev/null +++ b/examples/18.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:516b41acac2c03f71c67179a606c2db44f4259909d21a6b87720df34a6dfb22d +size 207267 diff --git a/examples/18_2LN8Wae.jpg b/examples/18_2LN8Wae.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8603d5c7329fafe873f2042c54fd576fef33871c --- /dev/null +++ b/examples/18_2LN8Wae.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:969f71388266d7f7ca33b33d0abba25b37ef4fe0226da0f53614d97b19b2317e +size 220324 diff --git a/examples/19.jpg b/examples/19.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5ebe1be6bc9c9af09f01f52f7392ae2a037e2e74 --- /dev/null +++ b/examples/19.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fe72e722eecc1f2b9fa2a141d14cb875062c3357253e9b462d72d53c7afe9bfd +size 144844 diff --git a/examples/19_bQOd8mo.jpg b/examples/19_bQOd8mo.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7f544a8df4441a189ed960acf92c7652679ba468 --- /dev/null +++ b/examples/19_bQOd8mo.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:155f9e8142a962fb1bb2fb74c63fe022def921b5e1e8f7f5a559566b7830fcfa +size 266785 diff --git a/examples/2.jpg b/examples/2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4c49ebd0bf53657b7fdc0f3fdd5f257ac853f218 --- /dev/null +++ b/examples/2.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e01b2fa5c4359ccdde7185baa6751543fce744ab4495f7612f62dadd444541b9 +size 178822 diff --git a/examples/20.jpg b/examples/20.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9faa362fda04ea40c27142992085c9b3b55a4a2d --- /dev/null +++ b/examples/20.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e932f87dd8eed222905932553ccf26ce3e9336ad5de376062df2d5f494291b55 +size 184914 diff --git a/examples/20_d4qEaSG.jpg b/examples/20_d4qEaSG.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b9633000eeaa714fe38a0852dfcc7b13cf445226 --- /dev/null +++ b/examples/20_d4qEaSG.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a44de5f99562697b60263e04b3c4f5889d4f6d89645baa5031cccdf082ada362 +size 224552 diff --git a/examples/21.jpg b/examples/21.jpg new file mode 100644 index 0000000000000000000000000000000000000000..da903ce138f3856e8c403e8b03c7d7d27fb952f9 --- /dev/null +++ b/examples/21.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:684c12d587d1792866372c0d3f0ee767957c08fa8d750910199108f0af94120c +size 186454 diff --git a/examples/21_E6oxZY3.jpg b/examples/21_E6oxZY3.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a5386259debadcf275092b88be453232f3711498 --- /dev/null +++ b/examples/21_E6oxZY3.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c11805872e7b4bad13523ffb31e84efe73a21b5dbc036c935bd2d3415d95ff22 +size 269937 diff --git a/examples/22.jpg b/examples/22.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ad6f5e6337e875979cea16fd2dfe96dfbbf6f5cd --- /dev/null +++ b/examples/22.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9056a6d98416c60b6b2f4fc8ecee27014ffc776a8ee52cec824d1252d189e271 +size 214975 diff --git a/examples/22_eimzU9L.jpg b/examples/22_eimzU9L.jpg new file mode 100644 index 0000000000000000000000000000000000000000..365bbd7adf312302b9804a6d521d3e1755771f67 --- /dev/null +++ b/examples/22_eimzU9L.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ebda8051e4ca062ee846180726786c9edbf78242ec96bdbdf8cc09713b2a45c4 +size 195279 diff --git a/examples/23.jpg b/examples/23.jpg new file mode 100644 index 0000000000000000000000000000000000000000..31071f3c18c5b474a8a833f24854dbf275e57262 --- /dev/null +++ b/examples/23.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:26e6f8c865697150625bb4627643cf6bc2c244c1865ddbe29298ad7b30676aec +size 161732 diff --git a/examples/23_XH0MPmx.jpg b/examples/23_XH0MPmx.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d55074f7ced5c11a56d0e3113202a8da2e156b01 --- /dev/null +++ b/examples/23_XH0MPmx.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5dc78bbe7977b7bb806bf2af75982893c2b1f662f5130aec3098022bd2797e8c +size 165118 diff --git a/examples/24.jpg b/examples/24.jpg new file mode 100644 index 0000000000000000000000000000000000000000..3a3b170d12dacfd986c3c3a3b5aadfafa6b63d45 --- /dev/null +++ b/examples/24.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a73dc1b9b6501687cf3a5ba4395d91e215b9c7bdd79630777872acace80eb63e +size 203803 diff --git a/examples/24_6gMkQPj.jpg b/examples/24_6gMkQPj.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ad39e257901c12636f4fc49ce4b104de831ebaf6 --- /dev/null +++ b/examples/24_6gMkQPj.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:96adf2e32916eff1fcabe38f3c893d48f7de0a41c94eae05609141a439f07169 +size 207787 diff --git a/examples/25.jpg b/examples/25.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f387b8eef1c56d89f25cb816ee1d9cb787ad6883 --- /dev/null +++ b/examples/25.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e5ce2898254605a29302e3620fe6bbd539728d0dc5aa9080eb15807fe09d830f +size 201570 diff --git a/examples/25_owD4RvO.jpg b/examples/25_owD4RvO.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6f02af55ad28b3bafa54f0fc2bc1400b2d988406 --- /dev/null +++ b/examples/25_owD4RvO.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1c2da1dc0e1ae3f852909a19d39bf15670f0e7189e03ddcd27fc620966fa0f98 +size 248391 diff --git a/examples/26.jpg b/examples/26.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ea8f520f2b6e119265d70016cd2fb1bf8133a3b2 --- /dev/null +++ b/examples/26.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:489d889f948d67cc8e4adcb654e491ef1516c5fae253132008a8dfb450dc3667 +size 293674 diff --git a/examples/26_eB80I22.jpg b/examples/26_eB80I22.jpg new file mode 100644 index 0000000000000000000000000000000000000000..34c56d1b16d9b6873a3fc8ff43cbd54443e1bbfc Binary files /dev/null and b/examples/26_eB80I22.jpg differ diff --git a/examples/27.jpg b/examples/27.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b4ca48557a5b37b7ba96191e7082d48298b6b087 --- /dev/null +++ b/examples/27.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1901c4da54ccd1b47d52a154d150215bc751c38f06d5d8a842857965f9f85095 +size 120730 diff --git a/examples/27_X2pkXLB.jpg b/examples/27_X2pkXLB.jpg new file mode 100644 index 0000000000000000000000000000000000000000..50629144e6bab79ebdc070ba49e9a8c52d18e19a --- /dev/null +++ b/examples/27_X2pkXLB.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:42c5fbbe57be80454e89091bed686e1fd9594f88d87871bafa82d8d1cab0e478 +size 210651 diff --git a/examples/28.jpg b/examples/28.jpg new file mode 100644 index 0000000000000000000000000000000000000000..179c241c78441726522ca1e109482255fe7bca83 --- /dev/null +++ b/examples/28.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c4a41fac1f05aa51c34777459c019bb5a4a8e15901fcc8d893bef5d32023629 +size 169078 diff --git a/examples/28_FP3TIDL.jpg b/examples/28_FP3TIDL.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7ebf84999e3d4ab45d894505b8650b8a44be503d --- /dev/null +++ b/examples/28_FP3TIDL.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:489733c1534d26f5970450f72508614fddcc4785baffa62d43bea4d4a0c7c9bc +size 283774 diff --git a/examples/29.jpg b/examples/29.jpg new file mode 100644 index 0000000000000000000000000000000000000000..42d76d275f36c5f191664dc281714ef13ec58c38 --- /dev/null +++ b/examples/29.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e6004d3c4bca7b7ce4bc9d345d10ef4825ef3e7dfb0c673cc176556941b3d4fd +size 162595 diff --git a/examples/3.jpg b/examples/3.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ba71ec9810929393a65d0697d039a7a7093cf7b1 --- /dev/null +++ b/examples/3.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c14219c94fcefe60d310a0eae2f16ef70efa06013ef67937e57e19f01f0c8504 +size 178725 diff --git a/examples/30.jpg b/examples/30.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4a59ed2c06b838e4e523dc024d455d717f02df20 --- /dev/null +++ b/examples/30.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:787417cc1b60ea36b7e277d1645faa8cf32fb536433f8dcd1c34d1cd35226bfd +size 116010 diff --git a/examples/4.jpg b/examples/4.jpg new file mode 100644 index 0000000000000000000000000000000000000000..ff474b6dbd6c6fe9df77880fa58651559d92afa8 --- /dev/null +++ b/examples/4.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:31f6d3aa6102013476c974eeef246c47083b5659eedafefc5d9429be75a16e62 +size 157489 diff --git a/examples/5.jpg b/examples/5.jpg new file mode 100644 index 0000000000000000000000000000000000000000..1a3e451a91f795fc79ddd4abfd49bf26ee4b674d --- /dev/null +++ b/examples/5.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:110aead585404a50638b7938575be0faf6e225c7c026f130af64cbb59a01e6b8 +size 296684 diff --git a/examples/6.jpg b/examples/6.jpg new file mode 100644 index 0000000000000000000000000000000000000000..be3923846fe32a7e87d2c1ec16656fe828adc089 --- /dev/null +++ b/examples/6.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2cc903e9ecedf034eb61bf66877e2dcb139744f6218c600573f1c81cdfd39895 +size 140868 diff --git a/examples/7.jpg b/examples/7.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f91b0c132d4070e4c2b4458267458c7c7b4682ba Binary files /dev/null and b/examples/7.jpg differ diff --git a/examples/8.jpg b/examples/8.jpg new file mode 100644 index 0000000000000000000000000000000000000000..0871f779716987366eee09fab0a0f3dd56adfa1b --- /dev/null +++ b/examples/8.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4368dcc15fdee5be71a5f1eb87ee1648f6103f03ded8e8943d3ca66620435eed +size 241082 diff --git a/examples/9.jpg b/examples/9.jpg new file mode 100644 index 0000000000000000000000000000000000000000..fc123019d4b80d32e8f57e108631181f9af36c70 --- /dev/null +++ b/examples/9.jpg @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:899a6d7629bf9a10e70b46703e1188d2beb696456b37fc115e014cc8f018987a +size 293143 diff --git a/metrics/BoxF1_curve.png b/metrics/BoxF1_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..dcd81688d7c1d127840e4ce0a52f5e1d2458b461 --- /dev/null +++ b/metrics/BoxF1_curve.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:af220fd82f80af92dab57aa567b5b1915a5e38f05f7bde854273faec7d9d32ff +size 539646 diff --git a/metrics/BoxPR_curve.png b/metrics/BoxPR_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..851cf57e6bd0cfd07960099884fb93384258187f --- /dev/null +++ b/metrics/BoxPR_curve.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:30c4c82f5d7a5bd459b1641e91986c107399811e8b9f2bc5f3684721eb353dc1 +size 223666 diff --git a/metrics/BoxP_curve.png b/metrics/BoxP_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..2491740b845ef5eb1817e4d35d7f7ab7fa3fdb1f --- /dev/null +++ b/metrics/BoxP_curve.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f2e9c28f87cd3aaabd140605dcfec75a64a9d69b2a70c48e92181240add1394 +size 307029 diff --git a/metrics/BoxR_curve.png b/metrics/BoxR_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..a89e5792bdcce79df3cc476d409938e26d0d4735 --- /dev/null +++ b/metrics/BoxR_curve.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8bf14bb996b3dbd75f9fc5cce0cca96f62a7660bde4331b60c6b95b9a7d610dc +size 499037 diff --git a/metrics/results.png b/metrics/results.png new file mode 100644 index 0000000000000000000000000000000000000000..5676e6b68982fd1e2f5119c25987676024c0e933 --- /dev/null +++ b/metrics/results.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b57d075e0629f7013dc1137b95dabe4088a858d12d70f24ddd32fdbc36bfc3bc +size 270591 diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000000000000000000000000000000000000..b4ac6128aa32647961b18c303e6b8e17b60b57c7 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,7 @@ +gradio>=5.0.0 +ultralytics +opencv-python-headless +numpy +pillow +torch +torchvision \ No newline at end of file