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

updates

This commit is contained in:
Glenn Jocher 2020-01-16 13:25:18 -08:00
parent 75933e93a1
commit c0cde1edf0
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
utils/datasets.py
View File

@ -215,6 +215,12 @@ class LoadStreams: # multiple IP or RTSP cameras
thread.start() thread.start()
print('') # newline print('') # newline
# check for common shapes
s = np.stack([letterbox(x, new_shape=self.img_size)[0].shape for x in self.imgs], 0) # inference shapes
self.rect = np.unique(s, axis=0).shape[0] == 1 # rect inference if all shapes equal
if not self.rect:
print('WARNING: Different stream shapes detected. For optimal performance supply similarly-shaped streams.')
def update(self, index, cap): def update(self, index, cap):
# Read next stream frame in a daemon thread # Read next stream frame in a daemon thread
n = 0 n = 0
@ -239,7 +245,7 @@ class LoadStreams: # multiple IP or RTSP cameras
raise StopIteration raise StopIteration
# Letterbox # Letterbox
img = [letterbox(x, new_shape=self.img_size, interp=cv2.INTER_LINEAR)[0] for x in img0] img = [letterbox(x, new_shape=self.img_size, auto=self.rect, interp=cv2.INTER_LINEAR)[0] for x in img0]
# Stack # Stack
img = np.stack(img, 0) img = np.stack(img, 0)