Spaces:
Running
Running
zhang-ziang
commited on
Commit
·
dbc9aea
1
Parent(s):
ed2605d
default value
Browse files
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)
|