IliaLarchenko
commited on
Commit
•
8f14232
1
Parent(s):
1d02673
fixed bugs in config
Browse files- configs/augmentations.json +28 -39
- src/control.py +2 -2
configs/augmentations.json
CHANGED
@@ -175,8 +175,8 @@
|
|
175 |
0.25
|
176 |
],
|
177 |
"limits_list": [
|
178 |
-
0,
|
179 |
-
|
180 |
],
|
181 |
"param_name": [
|
182 |
"scale_min",
|
@@ -294,21 +294,21 @@
|
|
294 |
{
|
295 |
"defaults": 5,
|
296 |
"limits_list": [
|
297 |
-
|
298 |
15
|
299 |
],
|
300 |
-
"param_name": "
|
301 |
"type": "num_interval"
|
302 |
},
|
303 |
{
|
304 |
"defaults": [
|
305 |
-
-2.0,
|
306 |
-
2.0
|
307 |
-
],
|
308 |
-
"limits_list": [
|
309 |
-0.3,
|
310 |
0.3
|
311 |
],
|
|
|
|
|
|
|
|
|
312 |
"param_name": "distort_limit",
|
313 |
"type": "num_interval"
|
314 |
},
|
@@ -469,20 +469,6 @@
|
|
469 |
"type": "radio"
|
470 |
}
|
471 |
],
|
472 |
-
"MedianBlur": [
|
473 |
-
{
|
474 |
-
"defaults": [
|
475 |
-
3,
|
476 |
-
7
|
477 |
-
],
|
478 |
-
"limits_list": [
|
479 |
-
3,
|
480 |
-
100
|
481 |
-
],
|
482 |
-
"param_name": "blur_limit",
|
483 |
-
"type": "num_interval"
|
484 |
-
}
|
485 |
-
],
|
486 |
"MotionBlur": [
|
487 |
{
|
488 |
"defaults": [
|
@@ -500,13 +486,13 @@
|
|
500 |
"MultiplicativeNoise": [
|
501 |
{
|
502 |
"defaults": [
|
503 |
-
0.1,
|
504 |
-
5.0
|
505 |
-
],
|
506 |
-
"limits_list": [
|
507 |
0.9,
|
508 |
1.1
|
509 |
],
|
|
|
|
|
|
|
|
|
510 |
"param_name": "multiplier",
|
511 |
"type": "num_interval"
|
512 |
},
|
@@ -706,8 +692,8 @@
|
|
706 |
],
|
707 |
"param_name": "grid",
|
708 |
"subparam_names": [
|
709 |
-
"
|
710 |
-
"
|
711 |
],
|
712 |
"type": "several_nums"
|
713 |
}
|
@@ -773,7 +759,7 @@
|
|
773 |
"None",
|
774 |
"drizzle",
|
775 |
"heavy",
|
776 |
-
"
|
777 |
],
|
778 |
"param_name": "rain_type",
|
779 |
"type": "radio"
|
@@ -868,7 +854,10 @@
|
|
868 |
{
|
869 |
"param_name": "min_max_height",
|
870 |
"placeholder": {
|
871 |
-
"defaults":
|
|
|
|
|
|
|
872 |
"limits_list": [
|
873 |
1,
|
874 |
"image_height"
|
@@ -902,7 +891,7 @@
|
|
902 |
"defaults": 1.0,
|
903 |
"limits_list": [
|
904 |
0.1,
|
905 |
-
|
906 |
],
|
907 |
"param_name": "w2h_ratio",
|
908 |
"type": "num_interval"
|
@@ -1023,25 +1012,25 @@
|
|
1023 |
"ShiftScaleRotate": [
|
1024 |
{
|
1025 |
"defaults": [
|
1026 |
-
-1.0,
|
1027 |
-
1.0
|
1028 |
-
],
|
1029 |
-
"limits_list": [
|
1030 |
-0.06,
|
1031 |
0.06
|
1032 |
],
|
|
|
|
|
|
|
|
|
1033 |
"param_name": "shift_limit",
|
1034 |
"type": "num_interval"
|
1035 |
},
|
1036 |
{
|
1037 |
"defaults": [
|
1038 |
-
-2.0,
|
1039 |
-
2.0
|
1040 |
-
],
|
1041 |
-
"limits_list": [
|
1042 |
-0.1,
|
1043 |
0.1
|
1044 |
],
|
|
|
|
|
|
|
|
|
1045 |
"param_name": "scale_limit",
|
1046 |
"type": "num_interval"
|
1047 |
},
|
|
|
175 |
0.25
|
176 |
],
|
177 |
"limits_list": [
|
178 |
+
0.01,
|
179 |
+
0.99
|
180 |
],
|
181 |
"param_name": [
|
182 |
"scale_min",
|
|
|
294 |
{
|
295 |
"defaults": 5,
|
296 |
"limits_list": [
|
297 |
+
1,
|
298 |
15
|
299 |
],
|
300 |
+
"param_name": "num_steps",
|
301 |
"type": "num_interval"
|
302 |
},
|
303 |
{
|
304 |
"defaults": [
|
|
|
|
|
|
|
|
|
305 |
-0.3,
|
306 |
0.3
|
307 |
],
|
308 |
+
"limits_list": [
|
309 |
+
-2.0,
|
310 |
+
2.0
|
311 |
+
],
|
312 |
"param_name": "distort_limit",
|
313 |
"type": "num_interval"
|
314 |
},
|
|
|
469 |
"type": "radio"
|
470 |
}
|
471 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
472 |
"MotionBlur": [
|
473 |
{
|
474 |
"defaults": [
|
|
|
486 |
"MultiplicativeNoise": [
|
487 |
{
|
488 |
"defaults": [
|
|
|
|
|
|
|
|
|
489 |
0.9,
|
490 |
1.1
|
491 |
],
|
492 |
+
"limits_list": [
|
493 |
+
0.1,
|
494 |
+
5.0
|
495 |
+
],
|
496 |
"param_name": "multiplier",
|
497 |
"type": "num_interval"
|
498 |
},
|
|
|
692 |
],
|
693 |
"param_name": "grid",
|
694 |
"subparam_names": [
|
695 |
+
"vertical",
|
696 |
+
"horizontal"
|
697 |
],
|
698 |
"type": "several_nums"
|
699 |
}
|
|
|
759 |
"None",
|
760 |
"drizzle",
|
761 |
"heavy",
|
762 |
+
"torrential"
|
763 |
],
|
764 |
"param_name": "rain_type",
|
765 |
"type": "radio"
|
|
|
854 |
{
|
855 |
"param_name": "min_max_height",
|
856 |
"placeholder": {
|
857 |
+
"defaults": [
|
858 |
+
"image_half_height",
|
859 |
+
"image_height"
|
860 |
+
],
|
861 |
"limits_list": [
|
862 |
1,
|
863 |
"image_height"
|
|
|
891 |
"defaults": 1.0,
|
892 |
"limits_list": [
|
893 |
0.1,
|
894 |
+
1.0
|
895 |
],
|
896 |
"param_name": "w2h_ratio",
|
897 |
"type": "num_interval"
|
|
|
1012 |
"ShiftScaleRotate": [
|
1013 |
{
|
1014 |
"defaults": [
|
|
|
|
|
|
|
|
|
1015 |
-0.06,
|
1016 |
0.06
|
1017 |
],
|
1018 |
+
"limits_list": [
|
1019 |
+
-1.0,
|
1020 |
+
1.0
|
1021 |
+
],
|
1022 |
"param_name": "shift_limit",
|
1023 |
"type": "num_interval"
|
1024 |
},
|
1025 |
{
|
1026 |
"defaults": [
|
|
|
|
|
|
|
|
|
1027 |
-0.1,
|
1028 |
0.1
|
1029 |
],
|
1030 |
+
"limits_list": [
|
1031 |
+
-2.0,
|
1032 |
+
2.0
|
1033 |
+
],
|
1034 |
"param_name": "scale_limit",
|
1035 |
"type": "num_interval"
|
1036 |
},
|
src/control.py
CHANGED
@@ -63,13 +63,13 @@ def replace_none(string):
|
|
63 |
|
64 |
def select_radio(param_name, options_list, **kwargs):
|
65 |
st.sidebar.subheader(param_name)
|
66 |
-
result = st.sidebar.radio("", options_list)
|
67 |
return replace_none(result)
|
68 |
|
69 |
|
70 |
def select_checkbox(param_name, defaults, **kwargs):
|
71 |
st.sidebar.subheader(param_name)
|
72 |
-
result = st.sidebar.checkbox("True", defaults)
|
73 |
return result
|
74 |
|
75 |
|
|
|
63 |
|
64 |
def select_radio(param_name, options_list, **kwargs):
|
65 |
st.sidebar.subheader(param_name)
|
66 |
+
result = st.sidebar.radio("", options_list, key=hash(param_name))
|
67 |
return replace_none(result)
|
68 |
|
69 |
|
70 |
def select_checkbox(param_name, defaults, **kwargs):
|
71 |
st.sidebar.subheader(param_name)
|
72 |
+
result = st.sidebar.checkbox("True", defaults, key=hash(param_name))
|
73 |
return result
|
74 |
|
75 |
|