print("initializing...") windows = slide_window.completeScan(file) foundwindows = [] print("Processing",len(windows),"windows high...") for window in windows: wimage = image[window[0][1]:window[1][1], window[0][0]:window[1][0]] wfeatures = chog.extract_features(wimage, cspace='HLS', spatial_size=(32, 32), orient=orient, pix_per_cell=pix_per_cell, cell_per_block=cell_per_block, hist_bins=32, hist_range=(0, 256), hog_channel=1) if wfeatures is not None: confidence = svc.decision_function(wfeatures.reshape(1, -1)) if confidence[0] > hthreshold: foundwindows.append(window) window_img1 = chog.draw_boxes(image, foundwindows, color=(0, 0, 255), thick=2) foundwindows = [] print("Processing",len(windows),"windows low...") for window in windows: wimage = image[window[0][1]:window[1][1], window[0][0]:window[1][0]] wfeatures = chog.extract_features(wimage, cspace='HLS', spatial_size=(32, 32), orient=orient, pix_per_cell=pix_per_cell, cell_per_block=cell_per_block, hist_bins=32, hist_range=(0, 256), hog_channel=1) if wfeatures is not None: confidence = svc.decision_function(wfeatures.reshape(1, -1)) if confidence[0] > lthreshold: foundwindows.append(window) window_img2 = chog.draw_boxes(image, foundwindows, color=(0, 0, 255), thick=2) outimages.append((file, orient, pix_per_cell, cell_per_block, window_img1, window_img2))
import matplotlib.pyplot as plt import matplotlib.gridspec as gridspec import numpy as np import cv2 import glob import time from sklearn.svm import LinearSVC from sklearn.preprocessing import StandardScaler from skimage.feature import hog from sklearn.externals import joblib from testlib.slidingWindows import SlidingWindows from testlib.CHOG import CHOG versionName = 'CHOG-TEST-XO2' visualfile = './visualized/' + versionName + '-augmented.jpg' slide_window = SlidingWindows() chog = CHOG() outimages = [] images = glob.glob('./test_images/test*proj.jpg') for file in images: print("processing: ", file) image = cv2.cvtColor(cv2.imread(file), cv2.COLOR_BGR2RGB) print("initializing...") windows = slide_window.sentinalScan(file) window_img = chog.draw_boxes(image, windows, color=(0, 0, 255), thick=2) outimages.append((file, 0, 0, 0, image, window_img)) chog.drawXOPlots(visualfile, versionName, outimages)