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)