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 |
|