def find_hexagon(self, im):
res = images.crop_polygon(im)
crop = res.bbox
points = res.points
mask = res.mask
center = np.mean(points, axis=0)
return points, center, crop, mask
def manual(frame, parameters):
no_of_sides = parameters['no_of_sides']
if no_of_sides == 1:
crop_result = images.crop_circle(frame)
else:
crop_result = images.crop_polygon(frame)
return crop_result.bbox, crop_result.mask, crop_result.points
def get_crop_result(file):
vid = video.ReadVideo(file)
frame = vid.read_next_frame()
return images.crop_polygon(frame)
date = '090221'
n = 3
data_save = f"/media/data/Data/Orderphobic/TwoIntruders/OneIntruder/Logging/{date}_{D}_{n}.txt"
im_save = os.path.splitext(data_save)[0]
if not os.path.exists(im_save):
os.mkdir(im_save)
# cam_num = camera.guess_camera_number()
cam = camera.Camera(0)
frame = cam.get_frame()
if R != 0:
frame = ndimage.rotate(frame, R, reshape=False)
crop_result = images.crop_polygon(frame)
crop = crop_result.bbox
s = shaker.Shaker()
# s.ramp(650, D, 1)
s.change_duty(D)
def get_frame():
frame = cam.get_frame()
if R != 0:
frame = ndimage.rotate(frame, R, reshape=False)
frame = images.crop(frame, crop)
return frame
def get_circles(frame):
red = frame[:, :, 0] - frame[:, :, 2]
def get_crop(self):
frame = self.cam.get_frame()
crop_result = images.crop_polygon(frame)
return crop_result.bbox, crop_result.mask
def get_crops(self):
frame = self.get_frame(crop=False)
crop_result = images.crop_polygon(frame)
self.crop = crop_result.bbox
self.mask = crop_result.mask
frame = self.get_frame()
def get_both_bounds(frame):
left_crop_result = images.crop_polygon(frame)
right_crop_result = images.crop_polygon(frame)
return left_crop_result, right_crop_result
def get_crops(self):
frame = self.get_frame(crop=False)
crop_result = images.crop_polygon(frame)
self.crop = crop_result.bbox
frame = self.get_frame()
self.left_crop, self.right_crop = self.get_both_bounds(frame)
