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