fmajer commited on
Commit
c9830ba
·
1 Parent(s): 0b5d30d

border crop mode

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ config = resolve_data_config({}, model=vit)
35
  config['no_aug'] = True
36
  config['interpolation'] = 'bilinear'
37
  config['crop_pct'] = 1
38
- config['crop_mode'] = 'squash'
39
  transform = create_transform(**config)
40
 
41
  # Inference function
 
35
  config['no_aug'] = True
36
  config['interpolation'] = 'bilinear'
37
  config['crop_pct'] = 1
38
+ config['crop_mode'] = 'border'
39
  transform = create_transform(**config)
40
 
41
  # Inference function