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)
|