hamhanry commited on
Commit
31638be
·
1 Parent(s): 3d9d465

update: image size is set to 1024 by default

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -62,11 +62,11 @@ def app():
62
  model_id = gr.Dropdown(
63
  label="Model",
64
  choices=[
65
- "yolov10n-obb-obb",
66
- "yolov10s-obb-obb",
67
- "yolov10m-obb-bb",
68
  ],
69
- value="yolov10n-obb-obb",
70
  )
71
  image_size = gr.Slider(
72
  label="Image Size",
@@ -120,37 +120,37 @@ def app():
120
  [
121
  "test_images/P0024.jpg",
122
  "yolov10n-obb",
123
- 640,
124
  0.25,
125
  ],
126
  [
127
  "test_images/P0035.jpg",
128
  "yolov10n-obb",
129
- 640,
130
  0.25,
131
  ],
132
  [
133
  "test_images/P00121.jpg",
134
  "yolov10n-obb",
135
- 640,
136
  0.25,
137
  ],
138
  [
139
  "test_images/P0180.jpg",
140
  "yolov10n-obb",
141
- 640,
142
  0.25,
143
  ],
144
  [
145
  "test_images/P0279.jpg",
146
  "yolov10n-obb",
147
- 640,
148
  0.25,
149
  ],
150
  [
151
  "test_images/P2112.jpg",
152
  "yolov10n-obb",
153
- 640,
154
  0.25,
155
  ],
156
  ],
 
62
  model_id = gr.Dropdown(
63
  label="Model",
64
  choices=[
65
+ "yolov10n-obb",
66
+ "yolov10s-obb",
67
+ "yolov10m-obb",
68
  ],
69
+ value="yolov10n-obb",
70
  )
71
  image_size = gr.Slider(
72
  label="Image Size",
 
120
  [
121
  "test_images/P0024.jpg",
122
  "yolov10n-obb",
123
+ 1024,
124
  0.25,
125
  ],
126
  [
127
  "test_images/P0035.jpg",
128
  "yolov10n-obb",
129
+ 1024,
130
  0.25,
131
  ],
132
  [
133
  "test_images/P00121.jpg",
134
  "yolov10n-obb",
135
+ 1024,
136
  0.25,
137
  ],
138
  [
139
  "test_images/P0180.jpg",
140
  "yolov10n-obb",
141
+ 1024,
142
  0.25,
143
  ],
144
  [
145
  "test_images/P0279.jpg",
146
  "yolov10n-obb",
147
+ 1024,
148
  0.25,
149
  ],
150
  [
151
  "test_images/P2112.jpg",
152
  "yolov10n-obb",
153
+ 1024,
154
  0.25,
155
  ],
156
  ],