Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates 

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
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) elif arc == 'Fdefault': # Focal default no pw (28 cls, 21 obj, no pw)
b = [-2.1, -1.8] b = [-2.1, -1.8]
elif arc == 'uFBCE': # unified FocalBCE (5120 obj, 80 classes) 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) elif arc == 'uFCE': # unified FocalCE (64 cls, 1 background + 80 classes)
b = [7, -0.1] b = [7, -0.1]

1
utils/datasets.py
View File

@ -146,6 +146,7 @@ class LoadWebcam: # for inference
def __next__(self): def __next__(self):
self.count += 1 self.count += 1
if cv2.waitKey(1) == 27: # esc to quit if cv2.waitKey(1) == 27: # esc to quit
self.cap.release()
cv2.destroyAllWindows() cv2.destroyAllWindows()
raise StopIteration raise StopIteration