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

updates

This commit is contained in:
Glenn Jocher 2019-08-11 17:47:44 +02:00
parent 4e9a8661b2
commit dfa999455f
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
utils/datasets.py
View File

@ -120,9 +120,13 @@ class LoadWebcam: # for inference
def __init__(self, img_size=416, half=False):
self.img_size = img_size
self.half = half # half precision fp16 images
self.cam = cv2.VideoCapture(0) # local camera
# self.cam = cv2.VideoCapture('rtsp://192.168.1.64/1') # IP camera
# self.cam = cv2.VideoCapture('rtsp://username:password@192.168.1.64/1') # IP camera with login
pipe = 0 # local camera
# pipe = 'rtsp://192.168.1.64/1' # IP camera
# pipe = 'rtsp://username:password@192.168.1.64/1' # IP camera with login
# pipe = '"rtspsrc location="rtsp://username:password@192.168.1.64/1" latency=10 ! appsink' # GStreamer https://answers.opencv.org/question/215996/changing-gstreamer-pipeline-to-opencv-in-pythonsolved/
self.cap = cv2.VideoCapture(pipe) # video capture object
def __iter__(self):
self.count = -1
@ -135,7 +139,7 @@ class LoadWebcam: # for inference
raise StopIteration
# Read image
ret_val, img0 = self.cam.read()
ret_val, img0 = self.cap.read()
assert ret_val, 'Webcam Error'
img_path = 'webcam_%g.jpg' % self.count
img0 = cv2.flip(img0, 1) # flip left-right
@ -156,7 +160,8 @@ class LoadWebcam: # for inference
class LoadImagesAndLabels(Dataset): # for training/testing
def __init__(self, path, img_size=416, batch_size=16, augment=False, hyp=None, rect=True, image_weights=False, cache_images=False):
def __init__(self, path, img_size=416, batch_size=16, augment=False, hyp=None, rect=True, image_weights=False,
cache_images=False):
path = str(Path(path)) # os-agnostic
with open(path, 'r') as f:
self.img_files = [x.replace('/', os.sep) for x in f.read().splitlines() # os-agnostic

2
utils/utils.py
View File

@ -775,7 +775,7 @@ def plot_results(start=0, stop=0): # from utils.utils import *; plot_results()
ax[i].get_shared_y_axes().join(ax[i], ax[i - 5])
fig.tight_layout()
ax[4].legend()
ax[0].legend()
fig.savefig('results.png', dpi=200)