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 = []
|