Spaces:
Sleeping
Sleeping
Commit
·
ba0e7f4
1
Parent(s):
3012224
Changed argument name
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def inference(image, conf_thres, iou_thres,):
|
|
18 |
pred = model(image, device=device, retina_masks=True, imgsz=1024, conf=conf_thres, iou=iou_thres)
|
19 |
prompt_process = FastSAMPrompt(input, pred, device="cpu")
|
20 |
ann = prompt_process.everything_prompt()
|
21 |
-
prompt_process.plot(annotations=ann,
|
22 |
output = Image.open('./output.jpg')
|
23 |
output = np.array(output)
|
24 |
return output
|
|
|
18 |
pred = model(image, device=device, retina_masks=True, imgsz=1024, conf=conf_thres, iou=iou_thres)
|
19 |
prompt_process = FastSAMPrompt(input, pred, device="cpu")
|
20 |
ann = prompt_process.everything_prompt()
|
21 |
+
prompt_process.plot(annotations=ann, output_path="./output.jpg", withContours=False, better_quality=False)
|
22 |
output = Image.open('./output.jpg')
|
23 |
output = np.array(output)
|
24 |
return output
|