zhang-ziang commited on
Commit
dbc9aea
·
1 Parent(s): ed2605d

default value

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ dino = dino.to(device)
29
  print('weight loaded')
30
  val_preprocess = AutoImageProcessor.from_pretrained(DINO_LARGE, cache_dir='./')
31
 
32
- def infer_func(img, do_rm_bkg, do_infer_aug):
33
  origin_img = Image.fromarray(img)
34
  if do_infer_aug:
35
  rm_bkg_img = background_preprocess(origin_img, True)
 
29
  print('weight loaded')
30
  val_preprocess = AutoImageProcessor.from_pretrained(DINO_LARGE, cache_dir='./')
31
 
32
+ def infer_func(img, do_rm_bkg=True, do_infer_aug=False):
33
  origin_img = Image.fromarray(img)
34
  if do_infer_aug:
35
  rm_bkg_img = background_preprocess(origin_img, True)