restoring to defualt
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ def infer(in_pil_img, in_threshold=0.8, out_pil_img=None):
|
|
| 77 |
|
| 78 |
# Run HMR2.0 on all detected humans
|
| 79 |
dataset = ViTDetDataset(model_cfg, open_cv_image, boxes)
|
| 80 |
-
dataloader = torch.utils.data.DataLoader(dataset, batch_size=
|
| 81 |
|
| 82 |
all_verts = []
|
| 83 |
all_cam_t = []
|
|
|
|
| 77 |
|
| 78 |
# Run HMR2.0 on all detected humans
|
| 79 |
dataset = ViTDetDataset(model_cfg, open_cv_image, boxes)
|
| 80 |
+
dataloader = torch.utils.data.DataLoader(dataset, batch_size=8, shuffle=False, num_workers=0)
|
| 81 |
|
| 82 |
all_verts = []
|
| 83 |
all_cam_t = []
|