Spaces:
Runtime error
Runtime error
AAAAAAyq
commited on
Commit
·
c2bd1fd
1
Parent(s):
1472255
Add more cached examples
Browse files
app.py
CHANGED
@@ -157,7 +157,7 @@ def fast_show_mask_gpu(annotation, ax,
|
|
157 |
|
158 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
159 |
|
160 |
-
def predict(input, input_size=512, high_visual_quality=
|
161 |
input_size = int(input_size) # 确保 imgsz 是整数
|
162 |
results = model(input, device=device, retina_masks=True, iou=0.7, conf=0.25, imgsz=input_size)
|
163 |
fig = fast_process(annotations=results[0].masks.data,
|
|
|
157 |
|
158 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
159 |
|
160 |
+
def predict(input, input_size=512, high_visual_quality=True):
|
161 |
input_size = int(input_size) # 确保 imgsz 是整数
|
162 |
results = model(input, device=device, retina_masks=True, iou=0.7, conf=0.25, imgsz=input_size)
|
163 |
fig = fast_process(annotations=results[0].masks.data,
|