Intial Commit

.gitignore

@@ -0,0 +1,3 @@
+flagged/
+*.pt
+gradio_cached_examples/

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+import torch
+from PIL import Image
+
+import torch
+
+# Model
+# model = torch.hub.load("ultralytics/yolov5", "yolov5s")  # or yolov5n - yolov5x6, custom
+path = "obj.pt"
+model = torch.hub.load('ultralytics/yolov5', 'custom', path=path)
+
+
+def yolo(im, size=640):
+
+    g = (size / max(im.size))  # gain
+
+    im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
+
+    results = model(im)  # inference
+
+    # results.render()  # updates results.imgs with boxes and labels
+
+    # return Image.fromarray(results.imgs[0])
+
+    # Retrieve the annotated image from the results (modify this based on your model's output structure)
+    annotated_image = results.render()[0] 
+
+    # Return the annotated image
+    return Image.fromarray(annotated_image)
+
+inputs = gr.inputs.Image(type='pil', label="Original Image")
+outputs = gr.outputs.Image(type="filepath", label="Output Image")
+
+
+title = "YOLOv5"
+description = "YOLOv5 Gradio demo for object detection. Upload an image or click an example image to use."
+
+article = "YOLOv5 Leprosy AI Demo" 
+
+examples = [["lp1.jpg"],["lp2.jpg"],["nlp1.jpg"],["nlp2.jpg"]]
+gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, analytics_enabled=False).launch(
+    debug=True)

requirements.txt

@@ -0,0 +1,51 @@
+# YOLOv5 requirements
+# Usage: pip install -r requirements.txt
+
+# Base ------------------------------------------------------------------------
+gitpython>=3.1.30
+matplotlib>=3.3
+numpy>=1.22.2
+opencv-python>=4.1.1
+Pillow>=7.1.2
+psutil  # system resources
+PyYAML>=5.3.1
+requests>=2.23.0
+scipy>=1.4.1
+thop>=0.1.1  # FLOPs computation
+torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
+torchvision>=0.9.0
+tqdm>=4.64.0
+ultralytics>=8.0.147
+# protobuf<=3.20.1  # https://github.com/ultralytics/yolov5/issues/8012
+
+# Logging ---------------------------------------------------------------------
+# tensorboard>=2.4.1
+# clearml>=1.2.0
+# comet
+
+# Plotting --------------------------------------------------------------------
+pandas>=1.1.4
+seaborn>=0.11.0
+
+# Export ----------------------------------------------------------------------
+# coremltools>=6.0  # CoreML export
+# onnx>=1.10.0  # ONNX export
+# onnx-simplifier>=0.4.1  # ONNX simplifier
+# nvidia-pyindex  # TensorRT export
+# nvidia-tensorrt  # TensorRT export
+# scikit-learn<=1.1.2  # CoreML quantization
+# tensorflow>=2.4.0  # TF exports (-cpu, -aarch64, -macos)
+# tensorflowjs>=3.9.0  # TF.js export
+# openvino-dev>=2023.0  # OpenVINO export
+
+# Deploy ----------------------------------------------------------------------
+setuptools>=65.5.1 # Snyk vulnerability fix
+# tritonclient[all]~=2.24.0
+
+# Extras ----------------------------------------------------------------------
+# ipython  # interactive notebook
+# mss  # screenshots
+# albumentations>=1.0.3
+# pycocotools>=2.0.6  # COCO mAP
+
+gradio