IliaLarchenko
commited on
Commit
•
75161a8
1
Parent(s):
6684d9c
refactored some names
Browse files- configs/augmentations.json +27 -27
- src/control.py +5 -5
configs/augmentations.json
CHANGED
@@ -10,7 +10,7 @@
|
|
10 |
100
|
11 |
],
|
12 |
"param_name": "blur_limit",
|
13 |
-
"type": "
|
14 |
}
|
15 |
],
|
16 |
"CLAHE": [
|
@@ -24,7 +24,7 @@
|
|
24 |
100
|
25 |
],
|
26 |
"param_name": "clip_limit",
|
27 |
-
"type": "
|
28 |
},
|
29 |
{
|
30 |
"defaults_list": [
|
@@ -46,7 +46,7 @@
|
|
46 |
"height",
|
47 |
"width"
|
48 |
],
|
49 |
-
"type": "
|
50 |
}
|
51 |
],
|
52 |
"CenterCrop": [
|
@@ -59,7 +59,7 @@
|
|
59 |
"image_height"
|
60 |
]
|
61 |
},
|
62 |
-
"type": "
|
63 |
},
|
64 |
{
|
65 |
"param_name": "width",
|
@@ -70,7 +70,7 @@
|
|
70 |
"image_width"
|
71 |
]
|
72 |
},
|
73 |
-
"type": "
|
74 |
}
|
75 |
],
|
76 |
"ChannelDropout": [
|
@@ -84,7 +84,7 @@
|
|
84 |
3
|
85 |
],
|
86 |
"param_name": "channel_drop_range",
|
87 |
-
"type": "
|
88 |
},
|
89 |
{
|
90 |
"defaults": 0,
|
@@ -93,7 +93,7 @@
|
|
93 |
255
|
94 |
],
|
95 |
"param_name": "fill_value",
|
96 |
-
"type": "
|
97 |
}
|
98 |
],
|
99 |
"ChannelShuffle": [],
|
@@ -143,7 +143,7 @@
|
|
143 |
100
|
144 |
],
|
145 |
"param_name": "num_holes",
|
146 |
-
"type": "
|
147 |
},
|
148 |
{
|
149 |
"defaults": 8,
|
@@ -152,7 +152,7 @@
|
|
152 |
100
|
153 |
],
|
154 |
"param_name": "max_h_size",
|
155 |
-
"type": "
|
156 |
},
|
157 |
{
|
158 |
"defaults": 8,
|
@@ -161,7 +161,7 @@
|
|
161 |
100
|
162 |
],
|
163 |
"param_name": "max_w_size",
|
164 |
-
"type": "
|
165 |
},
|
166 |
{
|
167 |
"param_name": "fill_value",
|
@@ -195,7 +195,7 @@
|
|
195 |
500.0
|
196 |
],
|
197 |
"param_name": "var_limit",
|
198 |
-
"type": "
|
199 |
},
|
200 |
{
|
201 |
"defaults": 0.0,
|
@@ -204,7 +204,7 @@
|
|
204 |
100.0
|
205 |
],
|
206 |
"param_name": "mean",
|
207 |
-
"type": "
|
208 |
}
|
209 |
],
|
210 |
"HorizontalFlip": [],
|
@@ -219,7 +219,7 @@
|
|
219 |
100
|
220 |
],
|
221 |
"param_name": "hue_shift_limit",
|
222 |
-
"type": "
|
223 |
},
|
224 |
{
|
225 |
"defaults": [
|
@@ -231,7 +231,7 @@
|
|
231 |
100
|
232 |
],
|
233 |
"param_name": "sat_shift_limit",
|
234 |
-
"type": "
|
235 |
},
|
236 |
{
|
237 |
"defaults": [
|
@@ -243,7 +243,7 @@
|
|
243 |
100
|
244 |
],
|
245 |
"param_name": "val_shift_limit",
|
246 |
-
"type": "
|
247 |
}
|
248 |
],
|
249 |
"ImageCompression": [
|
@@ -300,7 +300,7 @@
|
|
300 |
100
|
301 |
],
|
302 |
"param_name": "blur_limit",
|
303 |
-
"type": "
|
304 |
}
|
305 |
],
|
306 |
"MotionBlur": [
|
@@ -314,7 +314,7 @@
|
|
314 |
100
|
315 |
],
|
316 |
"param_name": "blur_limit",
|
317 |
-
"type": "
|
318 |
}
|
319 |
],
|
320 |
"RGBShift": [
|
@@ -328,7 +328,7 @@
|
|
328 |
255
|
329 |
],
|
330 |
"param_name": "r_shift_limit",
|
331 |
-
"type": "
|
332 |
},
|
333 |
{
|
334 |
"defaults": [
|
@@ -340,7 +340,7 @@
|
|
340 |
255
|
341 |
],
|
342 |
"param_name": "g_shift_limit",
|
343 |
-
"type": "
|
344 |
},
|
345 |
{
|
346 |
"defaults": [
|
@@ -352,7 +352,7 @@
|
|
352 |
255
|
353 |
],
|
354 |
"param_name": "b_shift_limit",
|
355 |
-
"type": "
|
356 |
}
|
357 |
],
|
358 |
"RandomBrightness": [
|
@@ -366,7 +366,7 @@
|
|
366 |
1.0
|
367 |
],
|
368 |
"param_name": "limit",
|
369 |
-
"type": "
|
370 |
}
|
371 |
],
|
372 |
"RandomContrast": [
|
@@ -380,7 +380,7 @@
|
|
380 |
1.0
|
381 |
],
|
382 |
"param_name": "limit",
|
383 |
-
"type": "
|
384 |
}
|
385 |
],
|
386 |
"RandomFog": [
|
@@ -406,7 +406,7 @@
|
|
406 |
1.0
|
407 |
],
|
408 |
"param_name": "alpha_coef",
|
409 |
-
"type": "
|
410 |
}
|
411 |
],
|
412 |
"RandomGridShuffle": [
|
@@ -430,7 +430,7 @@
|
|
430 |
"height",
|
431 |
"width"
|
432 |
],
|
433 |
-
"type": "
|
434 |
}
|
435 |
],
|
436 |
"RandomScale": [
|
@@ -444,7 +444,7 @@
|
|
444 |
2.0
|
445 |
],
|
446 |
"param_name": "scale_limit",
|
447 |
-
"type": "
|
448 |
},
|
449 |
{
|
450 |
"options_list": [
|
@@ -481,7 +481,7 @@
|
|
481 |
5.0
|
482 |
],
|
483 |
"param_name": "brightness_coeff",
|
484 |
-
"type": "
|
485 |
}
|
486 |
],
|
487 |
"Solarize": [
|
@@ -492,7 +492,7 @@
|
|
492 |
255
|
493 |
],
|
494 |
"param_name": "threshold",
|
495 |
-
"type": "
|
496 |
}
|
497 |
],
|
498 |
"ToGray": [],
|
|
|
10 |
100
|
11 |
],
|
12 |
"param_name": "blur_limit",
|
13 |
+
"type": "num_interval"
|
14 |
}
|
15 |
],
|
16 |
"CLAHE": [
|
|
|
24 |
100
|
25 |
],
|
26 |
"param_name": "clip_limit",
|
27 |
+
"type": "num_interval"
|
28 |
},
|
29 |
{
|
30 |
"defaults_list": [
|
|
|
46 |
"height",
|
47 |
"width"
|
48 |
],
|
49 |
+
"type": "several_nums"
|
50 |
}
|
51 |
],
|
52 |
"CenterCrop": [
|
|
|
59 |
"image_height"
|
60 |
]
|
61 |
},
|
62 |
+
"type": "num_interval"
|
63 |
},
|
64 |
{
|
65 |
"param_name": "width",
|
|
|
70 |
"image_width"
|
71 |
]
|
72 |
},
|
73 |
+
"type": "num_interval"
|
74 |
}
|
75 |
],
|
76 |
"ChannelDropout": [
|
|
|
84 |
3
|
85 |
],
|
86 |
"param_name": "channel_drop_range",
|
87 |
+
"type": "num_interval"
|
88 |
},
|
89 |
{
|
90 |
"defaults": 0,
|
|
|
93 |
255
|
94 |
],
|
95 |
"param_name": "fill_value",
|
96 |
+
"type": "num_interval"
|
97 |
}
|
98 |
],
|
99 |
"ChannelShuffle": [],
|
|
|
143 |
100
|
144 |
],
|
145 |
"param_name": "num_holes",
|
146 |
+
"type": "num_interval"
|
147 |
},
|
148 |
{
|
149 |
"defaults": 8,
|
|
|
152 |
100
|
153 |
],
|
154 |
"param_name": "max_h_size",
|
155 |
+
"type": "num_interval"
|
156 |
},
|
157 |
{
|
158 |
"defaults": 8,
|
|
|
161 |
100
|
162 |
],
|
163 |
"param_name": "max_w_size",
|
164 |
+
"type": "num_interval"
|
165 |
},
|
166 |
{
|
167 |
"param_name": "fill_value",
|
|
|
195 |
500.0
|
196 |
],
|
197 |
"param_name": "var_limit",
|
198 |
+
"type": "num_interval"
|
199 |
},
|
200 |
{
|
201 |
"defaults": 0.0,
|
|
|
204 |
100.0
|
205 |
],
|
206 |
"param_name": "mean",
|
207 |
+
"type": "num_interval"
|
208 |
}
|
209 |
],
|
210 |
"HorizontalFlip": [],
|
|
|
219 |
100
|
220 |
],
|
221 |
"param_name": "hue_shift_limit",
|
222 |
+
"type": "num_interval"
|
223 |
},
|
224 |
{
|
225 |
"defaults": [
|
|
|
231 |
100
|
232 |
],
|
233 |
"param_name": "sat_shift_limit",
|
234 |
+
"type": "num_interval"
|
235 |
},
|
236 |
{
|
237 |
"defaults": [
|
|
|
243 |
100
|
244 |
],
|
245 |
"param_name": "val_shift_limit",
|
246 |
+
"type": "num_interval"
|
247 |
}
|
248 |
],
|
249 |
"ImageCompression": [
|
|
|
300 |
100
|
301 |
],
|
302 |
"param_name": "blur_limit",
|
303 |
+
"type": "num_interval"
|
304 |
}
|
305 |
],
|
306 |
"MotionBlur": [
|
|
|
314 |
100
|
315 |
],
|
316 |
"param_name": "blur_limit",
|
317 |
+
"type": "num_interval"
|
318 |
}
|
319 |
],
|
320 |
"RGBShift": [
|
|
|
328 |
255
|
329 |
],
|
330 |
"param_name": "r_shift_limit",
|
331 |
+
"type": "num_interval"
|
332 |
},
|
333 |
{
|
334 |
"defaults": [
|
|
|
340 |
255
|
341 |
],
|
342 |
"param_name": "g_shift_limit",
|
343 |
+
"type": "num_interval"
|
344 |
},
|
345 |
{
|
346 |
"defaults": [
|
|
|
352 |
255
|
353 |
],
|
354 |
"param_name": "b_shift_limit",
|
355 |
+
"type": "num_interval"
|
356 |
}
|
357 |
],
|
358 |
"RandomBrightness": [
|
|
|
366 |
1.0
|
367 |
],
|
368 |
"param_name": "limit",
|
369 |
+
"type": "num_interval"
|
370 |
}
|
371 |
],
|
372 |
"RandomContrast": [
|
|
|
380 |
1.0
|
381 |
],
|
382 |
"param_name": "limit",
|
383 |
+
"type": "num_interval"
|
384 |
}
|
385 |
],
|
386 |
"RandomFog": [
|
|
|
406 |
1.0
|
407 |
],
|
408 |
"param_name": "alpha_coef",
|
409 |
+
"type": "num_interval"
|
410 |
}
|
411 |
],
|
412 |
"RandomGridShuffle": [
|
|
|
430 |
"height",
|
431 |
"width"
|
432 |
],
|
433 |
+
"type": "several_nums"
|
434 |
}
|
435 |
],
|
436 |
"RandomScale": [
|
|
|
444 |
2.0
|
445 |
],
|
446 |
"param_name": "scale_limit",
|
447 |
+
"type": "num_interval"
|
448 |
},
|
449 |
{
|
450 |
"options_list": [
|
|
|
481 |
5.0
|
482 |
],
|
483 |
"param_name": "brightness_coeff",
|
484 |
+
"type": "num_interval"
|
485 |
}
|
486 |
],
|
487 |
"Solarize": [
|
|
|
492 |
255
|
493 |
],
|
494 |
"param_name": "threshold",
|
495 |
+
"type": "num_interval"
|
496 |
}
|
497 |
],
|
498 |
"ToGray": [],
|
src/control.py
CHANGED
@@ -3,7 +3,7 @@ import streamlit as st
|
|
3 |
# TODO: rename and refactor everything
|
4 |
|
5 |
|
6 |
-
def
|
7 |
st.sidebar.subheader(param_name)
|
8 |
min_max_interval = st.sidebar.slider(
|
9 |
"", limits_list[0], limits_list[1], defaults, key=hash(param_name)
|
@@ -11,7 +11,7 @@ def select_int_interval(param_name: str, limits_list: list, defaults, **kwargs):
|
|
11 |
return min_max_interval
|
12 |
|
13 |
|
14 |
-
def
|
15 |
param_name, subparam_names, limits_list, defaults_list, **kwargs
|
16 |
):
|
17 |
st.sidebar.subheader(param_name)
|
@@ -45,7 +45,7 @@ def select_min_max(param_name, limits_list, defaults_list, min_diff=0, **kwargs)
|
|
45 |
|
46 |
|
47 |
def select_RGB(param_name, **kwargs):
|
48 |
-
result =
|
49 |
param_name,
|
50 |
subparam_names=["Red", "Green", "Blue"],
|
51 |
limits_list=[[0, 255], [0, 255], [0, 255]],
|
@@ -68,8 +68,8 @@ def select_checkbox(param_name, defaults, **kwargs):
|
|
68 |
|
69 |
# dict from param name to function showing this param
|
70 |
param2func = {
|
71 |
-
"
|
72 |
-
"
|
73 |
"radio": select_radio,
|
74 |
"rgb": select_RGB,
|
75 |
"checkbox": select_checkbox,
|
|
|
3 |
# TODO: rename and refactor everything
|
4 |
|
5 |
|
6 |
+
def select_num_interval(param_name: str, limits_list: list, defaults, **kwargs):
|
7 |
st.sidebar.subheader(param_name)
|
8 |
min_max_interval = st.sidebar.slider(
|
9 |
"", limits_list[0], limits_list[1], defaults, key=hash(param_name)
|
|
|
11 |
return min_max_interval
|
12 |
|
13 |
|
14 |
+
def select_several_nums(
|
15 |
param_name, subparam_names, limits_list, defaults_list, **kwargs
|
16 |
):
|
17 |
st.sidebar.subheader(param_name)
|
|
|
45 |
|
46 |
|
47 |
def select_RGB(param_name, **kwargs):
|
48 |
+
result = select_several_nums(
|
49 |
param_name,
|
50 |
subparam_names=["Red", "Green", "Blue"],
|
51 |
limits_list=[[0, 255], [0, 255], [0, 255]],
|
|
|
68 |
|
69 |
# dict from param name to function showing this param
|
70 |
param2func = {
|
71 |
+
"num_interval": select_num_interval,
|
72 |
+
"several_nums": select_several_nums,
|
73 |
"radio": select_radio,
|
74 |
"rgb": select_RGB,
|
75 |
"checkbox": select_checkbox,
|