IliaLarchenko commited on
Commit
8f14232
1 Parent(s): 1d02673

fixed bugs in config

Browse files
Files changed (2) hide show
  1. configs/augmentations.json +28 -39
  2. src/control.py +2 -2
configs/augmentations.json CHANGED
@@ -175,8 +175,8 @@
175
  0.25
176
  ],
177
  "limits_list": [
178
- 0,
179
- 1.0
180
  ],
181
  "param_name": [
182
  "scale_min",
@@ -294,21 +294,21 @@
294
  {
295
  "defaults": 5,
296
  "limits_list": [
297
- 0,
298
  15
299
  ],
300
- "param_name": "shift_limit",
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
- "height",
710
- "width"
711
  ],
712
  "type": "several_nums"
713
  }
@@ -773,7 +759,7 @@
773
  "None",
774
  "drizzle",
775
  "heavy",
776
- "torrestial"
777
  ],
778
  "param_name": "rain_type",
779
  "type": "radio"
@@ -868,7 +854,10 @@
868
  {
869
  "param_name": "min_max_height",
870
  "placeholder": {
871
- "defaults": "image_half_height",
 
 
 
872
  "limits_list": [
873
  1,
874
  "image_height"
@@ -902,7 +891,7 @@
902
  "defaults": 1.0,
903
  "limits_list": [
904
  0.1,
905
- 10.0
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