Update miner.py
Browse files
miner.py
CHANGED
|
@@ -1016,7 +1016,7 @@ HOMOGRAPHY_FILL_ONLY_VALID = True
|
|
| 1016 |
KP_THRESHOLD = 0.2 # new-5 style (was 0.3)
|
| 1017 |
# HRNet: smaller input = faster; 432x768 balances speed/accuracy (new-2 style)
|
| 1018 |
KP_H, KP_W = 360, 640
|
| 1019 |
-
HRNET_BATCH_SIZE =
|
| 1020 |
|
| 1021 |
|
| 1022 |
def _preprocess_batch(frames):
|
|
|
|
| 1016 |
KP_THRESHOLD = 0.2 # new-5 style (was 0.3)
|
| 1017 |
# HRNet: smaller input = faster; 432x768 balances speed/accuracy (new-2 style)
|
| 1018 |
KP_H, KP_W = 360, 640
|
| 1019 |
+
HRNET_BATCH_SIZE = 16 # larger batch = faster (if GPU mem allows)
|
| 1020 |
|
| 1021 |
|
| 1022 |
def _preprocess_batch(frames):
|