From b76962771e1ab227b4e36a515d174379009ee7f0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 Sep 2019 20:53:49 +0200 Subject: [PATCH] updates Signed-off-by: Glenn Jocher --- models.py | 2 +- utils/datasets.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/models.py b/models.py index 773e31a4..6982231c 100755 --- a/models.py +++ b/models.py @@ -88,7 +88,7 @@ def create_modules(module_defs, img_size, arc): elif arc == 'Fdefault': # Focal default no pw (28 cls, 21 obj, no pw) b = [-2.1, -1.8] elif arc == 'uFBCE': # unified FocalBCE (5120 obj, 80 classes) - b = [0, -3.5] + b = [0, -6.5] elif arc == 'uFCE': # unified FocalCE (64 cls, 1 background + 80 classes) b = [7, -0.1] diff --git a/utils/datasets.py b/utils/datasets.py index abf95658..166435b5 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -146,6 +146,7 @@ class LoadWebcam: # for inference def __next__(self): self.count += 1 if cv2.waitKey(1) == 27: # esc to quit + self.cap.release() cv2.destroyAllWindows() raise StopIteration