Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2019-09-02 20:53:49 +02:00
parent 0d5bf11fa5
commit b76962771e
2 changed files with 2 additions and 1 deletions

View File

@ -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]

View File

@ -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