def doit():
lane = lu.Lane(cd, pd, vwr=vwr, binary_warper=iu.hls_lab_lane_detect)
for path in ut.get_fnames("test_images/", "*.jpg"):
in_img = iu.imRead(path, reader='cv2', vwr=vwr)
out_img = lane.lane_finder_pipe(in_img, cd, pd, vwr=vwr)
ut.brk("check test_out")
def my_way(ploty, left_fit_cr, right_fit_cr, leftx, rightx):
cd = cache_dict
pd = parm_dict
path = ut.get_fnames("test_images/", "*.jpg")[0]
init_img, binary_warped = iu.get_binary_warped_image(path,
cd,
pd,
vwr=None)
# img is just to painlessly fake out Lane ctor
lane = lu.Lane(cd, pd, img=init_img, units='pixels', vwr=None)
ut.oneShotMsg("FIXME: units above must be meters")
lane.ploty = ploty
lane.left_bndry = lu.LaneBoundary(
0, # hope max ix doesnt matter for this test
binary_warped,
'L',
lane=lane,
vwr=None)
lane.right_bndry = lu.LaneBoundary(
0, # hope max ix doesnt matter for this test
binary_warped,
'R',
lane=lane,
vwr=None)
lane.left_bndry.x = leftx
lane.right_bndry.x = rightx
lane.left_bndry.fit_coeff = left_fit_cr
lane.right_bndry.fit_coeff = right_fit_cr
lane.left_bndry.radius_of_curvature()
lane.right_bndry.radius_of_curvature()
print("FIXME(meters): " +
str((lane.left_bndry.curve_radius, lane.right_bndry.curve_radius)))
def test_undistort(vwr, max_files):
for fname in ut.get_fnames("test_images/", "test*.jpg", max_files):
img = iu.imRead(fname, reader='cv2', vwr=vwr)
img = iu.undistort(img, cache_dict, vwr)
iv._push_deprecated(vwr,
img,
title="undist",
type="FIXME:shd come from img via imread")
def doit(path="", cd=None, pd=None, vwr=None):
for path in ut.get_fnames("test_images/", "*.jpg"):
lane = lu.Lane(cd, pd, vwr)
vwr.flush()
print("FIXME: path = %s" % path)
init_img, binary_warped = iu.get_binary_warped_image(path, cd, pd, vwr)
iv._push(vwr, init_img)
iv._push(vwr, binary_warped)
lane.find_pixels_all_bndrys(binary_warped)
lane.fit_polynomials()
vwr.show()
def doit(path="", cd=None, pd=None, vwr=None):
vwr.flush()
for path in ut.get_fnames("test_images/", "problem_binary_warped.jpg"):
img = iu.imRead(path,
flags=cv2.IMREAD_GRAYSCALE,
reader='cv2',
vwr=None)
ut.brk("wtf?")
hist = iu.hist(img)
ut.brk("plot the histogram")
ut.brk("call get_LR_hist_max_ix()")
iv._push(vwr, hist)
vwr.show()
def cb_corners(parm_dict, cache_dict, max_files=0, verbose=False, vwr=None):
# derived from same source as init_obj_points()
# termination criteria
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001)
nx, ny = parm_dict['chessboard_nx'], parm_dict['chessboard_ny']
objp = init_obj_points(nx, ny) # object points in 3 space
obj_points = []
img_points = [] # point in 2 space
iv._flush(vwr)
for fname in ut.get_fnames("camera_cal/", "calibration*.jpg"):
img_obj = imRead(fname, reader='cv2', vwr=vwr)
iv._push(vwr, img_obj)
if verbose:
print("cb_corners: fname = %s(%d, %d)" %
(fname, img_obj.shape()[0], img_obj.shape()[1]))
gray = cv2CvtColor(img_obj, cv2.COLOR_BGR2GRAY, vwr)
# Find the chessboard corners
ret, corners = cv2.findChessboardCorners(gray.img_data, (nx, ny), None)
# If found, add object points, image points
if ret == True:
obj_points.append(objp)
corners2 = cv2.cornerSubPix(gray.img_data, corners, (11, 11),
(-1, -1), criteria)
img_points.append(corners2)
# Draw and display the corners
cv2.drawChessboardCorners(img_obj.img_data, (nx, ny), corners2,
ret)
img_obj.title = img_obj.title + "_corners_drawn"
iv._push(vwr, img_obj)
else:
print("findChessboardCorners(%s) failed" % fname)
iv._show(vwr, clear=True)
cache_dict['obj_points'] = obj_points
cache_dict['img_points'] = img_points
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('13: ellipsys.py completed')
def main(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('5: conjunctions.py completed')
def main(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('4: prepositions.py completed')
def process_all(ifolder, ofolder):
main(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('10: cleanup.py completed')
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('11: fixRegular.py completed')
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('14: fixedDep.py completed')
def doit(path="", cd=None, pd=None, vwr=None):
vwr.flush()
for path in ut.get_fnames("test_images/", "*.jpg"):
img = iu.hls_lab_pipeline(path, cd, pd, vwr)
vwr.show()
def main(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('2: changeCompounds.py completed')
def process_all(ifolder, ofolder):
main(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('3: getCleanedCompounds.py completed')
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.tgt', '.xml'))
print('1: preparations.py completed')
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('15: punct.py completed')
def main(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('6: numerals.py completed')
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('12: checkReported.py completed')
def process_all(ifolder, ofolder):
munch(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('8: morphology.py completed')
def process_all(ifolder, ofolder):
main(*get_fnames(ifolder, ofolder, '.xml', '.xml'))
print('9: propernouns.py completed')
