IliaLarchenko
commited on
Commit
•
a99d36a
1
Parent(s):
a467828
fixed min_max config
Browse files- configs/augmentations.json +71 -62
- src/control.py +1 -1
configs/augmentations.json
CHANGED
@@ -202,40 +202,43 @@
|
|
202 |
],
|
203 |
"JpegCompression": [
|
204 |
{
|
205 |
-
"param_name": [
|
|
|
|
|
|
|
206 |
"type": "min_max",
|
207 |
"limits_list": [
|
208 |
0,
|
209 |
100
|
210 |
],
|
211 |
-
"defaults_list": [
|
|
|
|
|
|
|
212 |
}
|
213 |
],
|
214 |
"ImageCompression": [
|
215 |
{
|
216 |
-
"param_name": "
|
217 |
-
"type": "
|
218 |
-
"
|
219 |
0,
|
220 |
-
|
221 |
-
]
|
222 |
-
"defaults": 99
|
223 |
},
|
224 |
{
|
225 |
-
"param_name":
|
226 |
-
|
|
|
|
|
|
|
227 |
"limits_list": [
|
228 |
0,
|
229 |
100
|
230 |
],
|
231 |
-
"
|
232 |
-
|
233 |
-
|
234 |
-
"param_name": "compression_type",
|
235 |
-
"type": "radio",
|
236 |
-
"options_list": [
|
237 |
-
0,
|
238 |
-
1
|
239 |
]
|
240 |
}
|
241 |
],
|
@@ -310,51 +313,51 @@
|
|
310 |
],
|
311 |
"RandomSnow": [
|
312 |
{
|
313 |
-
"param_name":
|
314 |
-
|
315 |
-
|
316 |
-
0.0,
|
317 |
-
1.0
|
318 |
],
|
319 |
-
"
|
320 |
-
},
|
321 |
-
{
|
322 |
-
"param_name": "snow_point_upper",
|
323 |
-
"type": "int_interval",
|
324 |
"limits_list": [
|
325 |
0.0,
|
326 |
1.0
|
327 |
],
|
328 |
-
"
|
|
|
|
|
|
|
329 |
},
|
330 |
{
|
331 |
-
"param_name":
|
332 |
-
|
|
|
|
|
|
|
333 |
"limits_list": [
|
334 |
0.0,
|
335 |
20.0
|
336 |
],
|
337 |
-
"
|
|
|
|
|
|
|
338 |
}
|
339 |
],
|
340 |
"RandomFog": [
|
341 |
{
|
342 |
-
"param_name":
|
343 |
-
|
344 |
-
|
345 |
-
0.0,
|
346 |
-
1.0
|
347 |
],
|
348 |
-
"
|
349 |
-
},
|
350 |
-
{
|
351 |
-
"param_name": "fog_coef_upper",
|
352 |
-
"type": "int_interval",
|
353 |
"limits_list": [
|
354 |
0.0,
|
355 |
1.0
|
356 |
],
|
357 |
-
"
|
|
|
|
|
|
|
358 |
},
|
359 |
{
|
360 |
"param_name": "alpha_coef",
|
@@ -413,24 +416,30 @@
|
|
413 |
"param_name": "by_channels",
|
414 |
"type": "checkbox",
|
415 |
"defaults": 1
|
416 |
-
}
|
417 |
-
|
418 |
-
|
419 |
-
|
420 |
-
|
421 |
-
|
422 |
-
|
423 |
-
|
424 |
-
|
425 |
-
|
426 |
-
|
427 |
-
|
428 |
-
|
429 |
-
|
430 |
-
|
431 |
-
|
432 |
-
|
433 |
-
|
|
|
|
|
|
|
|
|
|
|
434 |
}
|
435 |
-
]
|
436 |
}
|
|
|
|
|
|
202 |
],
|
203 |
"JpegCompression": [
|
204 |
{
|
205 |
+
"param_name": [
|
206 |
+
"quality_lower",
|
207 |
+
"quality_upper"
|
208 |
+
],
|
209 |
"type": "min_max",
|
210 |
"limits_list": [
|
211 |
0,
|
212 |
100
|
213 |
],
|
214 |
+
"defaults_list": [
|
215 |
+
80,
|
216 |
+
100
|
217 |
+
]
|
218 |
}
|
219 |
],
|
220 |
"ImageCompression": [
|
221 |
{
|
222 |
+
"param_name": "compression_type",
|
223 |
+
"type": "radio",
|
224 |
+
"options_list": [
|
225 |
0,
|
226 |
+
1
|
227 |
+
]
|
|
|
228 |
},
|
229 |
{
|
230 |
+
"param_name": [
|
231 |
+
"quality_lower",
|
232 |
+
"quality_upper"
|
233 |
+
],
|
234 |
+
"type": "min_max",
|
235 |
"limits_list": [
|
236 |
0,
|
237 |
100
|
238 |
],
|
239 |
+
"defaults_list": [
|
240 |
+
80,
|
241 |
+
100
|
|
|
|
|
|
|
|
|
|
|
242 |
]
|
243 |
}
|
244 |
],
|
|
|
313 |
],
|
314 |
"RandomSnow": [
|
315 |
{
|
316 |
+
"param_name": [
|
317 |
+
"snow_point_lower",
|
318 |
+
"snow_point_upper"
|
|
|
|
|
319 |
],
|
320 |
+
"type": "min_max",
|
|
|
|
|
|
|
|
|
321 |
"limits_list": [
|
322 |
0.0,
|
323 |
1.0
|
324 |
],
|
325 |
+
"defaults_list": [
|
326 |
+
0.1,
|
327 |
+
0.2
|
328 |
+
]
|
329 |
},
|
330 |
{
|
331 |
+
"param_name": [
|
332 |
+
"brightness_lower",
|
333 |
+
"brightness_upper"
|
334 |
+
],
|
335 |
+
"type": "min_max",
|
336 |
"limits_list": [
|
337 |
0.0,
|
338 |
20.0
|
339 |
],
|
340 |
+
"defaults_list": [
|
341 |
+
2.5,
|
342 |
+
3.5
|
343 |
+
]
|
344 |
}
|
345 |
],
|
346 |
"RandomFog": [
|
347 |
{
|
348 |
+
"param_name": [
|
349 |
+
"fog_coef_lower",
|
350 |
+
"fog_coef_upper"
|
|
|
|
|
351 |
],
|
352 |
+
"type": "min_max",
|
|
|
|
|
|
|
|
|
353 |
"limits_list": [
|
354 |
0.0,
|
355 |
1.0
|
356 |
],
|
357 |
+
"defaults_list": [
|
358 |
+
0.1,
|
359 |
+
0.2
|
360 |
+
]
|
361 |
},
|
362 |
{
|
363 |
"param_name": "alpha_coef",
|
|
|
416 |
"param_name": "by_channels",
|
417 |
"type": "checkbox",
|
418 |
"defaults": 1
|
419 |
+
}
|
420 |
+
],
|
421 |
+
"CenterCrop": [
|
422 |
+
{
|
423 |
+
"param_name": "height",
|
424 |
+
"type": "int_interval",
|
425 |
+
"placeholder": {
|
426 |
+
"limits_list": [
|
427 |
+
1,
|
428 |
+
"image_height"
|
429 |
+
],
|
430 |
+
"defaults": "image_half_height"
|
431 |
+
}
|
432 |
+
},
|
433 |
+
{
|
434 |
+
"param_name": "width",
|
435 |
+
"type": "int_interval",
|
436 |
+
"placeholder": {
|
437 |
+
"limits_list": [
|
438 |
+
1,
|
439 |
+
"image_width"
|
440 |
+
],
|
441 |
+
"defaults": "image_half_width"
|
442 |
}
|
|
|
443 |
}
|
444 |
+
]
|
445 |
+
}
|
src/control.py
CHANGED
@@ -30,7 +30,7 @@ def select_several_ints(
|
|
30 |
|
31 |
def select_min_max(param_name, limits_list, defaults_list, **kwargs):
|
32 |
assert len(param_name) == 2
|
33 |
-
return select_int_interval('
|
34 |
|
35 |
|
36 |
def select_RGB(param_name, **kwargs):
|
|
|
30 |
|
31 |
def select_min_max(param_name, limits_list, defaults_list, **kwargs):
|
32 |
assert len(param_name) == 2
|
33 |
+
return select_int_interval(' & '.join(param_name), limits_list, defaults_list)
|
34 |
|
35 |
|
36 |
def select_RGB(param_name, **kwargs):
|