diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-18 18:38:31 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-18 18:38:31 +0800 |
| commit | 8012fce3e595aad31f4a52dc316b46e558234dff (patch) | |
| tree | 4a5e19a21ad8a4470931f5884777c127197153c0 /preprocess.py | |
| parent | 2988c1b9afd4e869b629a8629abedbf63d2452aa (diff) | |
| parent | 84a3d5991f2f7272d1be54ad6cfe6ce695f915a0 (diff) | |
Merge branch 'master' into data_parallel_py3.8
Diffstat (limited to 'preprocess.py')
| -rw-r--r-- | preprocess.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/preprocess.py b/preprocess.py index 894c7c1..eef59ba 100644 --- a/preprocess.py +++ b/preprocess.py @@ -52,7 +52,7 @@ def result_handler(frame_: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]: mask_x1 = (mask_xc + mask_half_width).ceil().int() # Skip incomplete frames - if (height < 64 or width < 64 / MASK_BOX_RATIO) \ + if (height < 64 or width < 64 / BOX_RATIO_THRESHOLD[1]) \ or (mask_x0 < 0 or mask_x1 > 320) \ or (mask_y0 < 0 or mask_y1 > 240): continue |