IliaLarchenko commited on
Commit
0c196bb
1 Parent(s): a99d36a

added crop

Browse files
Files changed (2) hide show
  1. configs/augmentations.json +38 -0
  2. src/control.py +11 -2
configs/augmentations.json CHANGED
@@ -441,5 +441,43 @@
441
  "defaults": "image_half_width"
442
  }
443
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
444
  ]
445
  }
 
441
  "defaults": "image_half_width"
442
  }
443
  }
444
+ ],
445
+ "Crop": [
446
+ {
447
+ "param_name": [
448
+ "x_min",
449
+ "x_max"
450
+ ],
451
+ "type": "min_max",
452
+ "placeholder": {
453
+ "limits_list": [
454
+ 0,
455
+ "image_width"
456
+ ],
457
+ "defaults_list": [
458
+ 0,
459
+ "image_half_width"
460
+ ]
461
+ },
462
+ "min_diff" : 1
463
+ },
464
+ {
465
+ "param_name": [
466
+ "y_min",
467
+ "y_max"
468
+ ],
469
+ "type": "min_max",
470
+ "placeholder": {
471
+ "limits_list": [
472
+ 0,
473
+ "image_height"
474
+ ],
475
+ "defaults_list": [
476
+ 0,
477
+ "image_half_height"
478
+ ]
479
+ },
480
+ "min_diff" : 1
481
+ }
482
  ]
483
  }
src/control.py CHANGED
@@ -28,9 +28,18 @@ def select_several_ints(
28
  return tuple(result)
29
 
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):
 
28
  return tuple(result)
29
 
30
 
31
+ def select_min_max(param_name, limits_list, defaults_list, min_diff=0, **kwargs):
32
  assert len(param_name) == 2
33
+ result = list(select_int_interval(" & ".join(param_name), limits_list, defaults_list))
34
+ if result[1] - result[0] < min_diff:
35
+ diff = min_diff - result[1] + result[0]
36
+ if result[1] + diff <= limits_list[1]:
37
+ result[1] = result[1] + diff
38
+ elif result[0] - diff >= limits_list[0]:
39
+ result[0] = result[0] - diff
40
+ else:
41
+ result = limits_list
42
+ return tuple(result)
43
 
44
 
45
  def select_RGB(param_name, **kwargs):