akbojda commited on
Commit
d199b74
·
1 Parent(s): e256e94

Add large model and more image examples

Browse files
Files changed (3) hide show
  1. app.py +6 -1
  2. examples/ex5.jpg +0 -0
  3. examples/ex6.jpg +0 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import yolov5
5
  models = {
6
  "yolov5s": yolov5.load("akbojda/yolov5s-aquarium"),
7
  "yolov5m": yolov5.load("akbojda/yolov5m-aquarium"),
 
8
  }
9
 
10
  def predict(img, model_type):
@@ -17,7 +18,9 @@ def predict(img, model_type):
17
  # Interface
18
  inputs = [
19
  gr.Image(),
20
- gr.Dropdown(["yolov5s", "yolov5m"], label="Model", value="yolov5s"),
 
 
21
  ]
22
 
23
  outputs = [
@@ -29,6 +32,8 @@ examples = [
29
  ["examples/ex2.jpg", None],
30
  ["examples/ex3.jpg", None],
31
  ["examples/ex4.jpg", None],
 
 
32
  ]
33
 
34
  title = "Aquarium object detection"
 
5
  models = {
6
  "yolov5s": yolov5.load("akbojda/yolov5s-aquarium"),
7
  "yolov5m": yolov5.load("akbojda/yolov5m-aquarium"),
8
+ "yolov5l": yolov5.load("akbojda/yolov5l-aquarium"),
9
  }
10
 
11
  def predict(img, model_type):
 
18
  # Interface
19
  inputs = [
20
  gr.Image(),
21
+ gr.Dropdown(["yolov5s", "yolov5m", "yolov5l"],
22
+ label="Model",
23
+ value="yolov5s"),
24
  ]
25
 
26
  outputs = [
 
32
  ["examples/ex2.jpg", None],
33
  ["examples/ex3.jpg", None],
34
  ["examples/ex4.jpg", None],
35
+ ["examples/ex5.jpg", None],
36
+ ["examples/ex6.jpg", None],
37
  ]
38
 
39
  title = "Aquarium object detection"
examples/ex5.jpg ADDED
examples/ex6.jpg ADDED