Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,8 +88,8 @@ def run_model(target_dir, model, anchor_size=100) -> dict:
|
|
| 88 |
# relocalization on all images
|
| 89 |
predictions_s = []
|
| 90 |
with tqdm(total=len(image_names), desc="Relocalizing") as pbar:
|
| 91 |
-
for img_split in images.split(
|
| 92 |
-
pbar.update(
|
| 93 |
predictions_s += to_cpu(
|
| 94 |
model.reloc(img_split, ret_img=True, memory_save=False)
|
| 95 |
)
|
|
|
|
| 88 |
# relocalization on all images
|
| 89 |
predictions_s = []
|
| 90 |
with tqdm(total=len(image_names), desc="Relocalizing") as pbar:
|
| 91 |
+
for img_split in images.split(15, dim=0):
|
| 92 |
+
pbar.update(15)
|
| 93 |
predictions_s += to_cpu(
|
| 94 |
model.reloc(img_split, ret_img=True, memory_save=False)
|
| 95 |
)
|