Upload 8 files

README.md

@@ -1,13 +1,12 @@
 ---
-title: IceVision
-emoji: 📈
-colorFrom: yellow
-colorTo: indigo
+title: Icevision_fridge_tutorial
+emoji: 🍼
+colorFrom: pink
+colorTo: purple
 sdk: gradio
-sdk_version: 4.1.0
 app_file: app.py
 pinned: false
-license: mit
+license: afl-3.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,46 @@
+from gradio.outputs import Label
+from icevision.all import *
+from icevision.models.checkpoint import *
+import PIL
+import gradio as gr
+import os
+
+# Load model
+checkpoint_path = "model_checkpoint.pth"
+checkpoint_and_model = model_from_checkpoint(checkpoint_path)
+model = checkpoint_and_model["model"]
+model_type = checkpoint_and_model["model_type"]
+class_map = checkpoint_and_model["class_map"]
+
+# Transforms
+img_size = checkpoint_and_model["img_size"]
+valid_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(img_size), tfms.A.Normalize()])
+
+# Populate examples in Gradio interface
+examples = [
+    ['./1.jpg'],
+    ['./2.jpg'],
+    ['./3.jpg']
+]
+
+def show_preds(input_image):
+    img = PIL.Image.fromarray(input_image, "RGB")
+    pred_dict = model_type.end2end_detect(img, valid_tfms, model, 
+                                          class_map=class_map, 
+                                          detection_threshold=0.5,
+                                          display_label=True, 
+                                          display_bbox=True, 
+                                          return_img=True, 
+                                          font_size=35,
+                                          label_color="#FF59D6")
+    return pred_dict["img"]
+
+gr_interface = gr.Interface(
+    fn=show_preds,
+    inputs=["image"],
+    outputs=[gr.outputs.Image(type="pil", label="VFNet Inference")],
+    title="Fridge Object Detector",
+    description="A VFNet model that detects common objects found in fridge. Upload an image or click an example image below to use.",
+    examples=examples,
+)
+gr_interface.launch(inline=False, share=False, debug=True)

gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

packages.txt

@@ -0,0 +1 @@
+python3-opencv

requirements.txt

@@ -0,0 +1,5 @@
+--find-links https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
+mmcv-full==1.3.17
+mmdet==2.17.0
+gradio==2.7.5
+icevision[all] == 0.12.0