Example #1
0
        def run(self, camera=False):
                frame = cv2.namedWindow(FRAME_NAME)

                # Set callback
                cv2.setMouseCallback(FRAME_NAME, self.draw)

                if camera:
                        cap = cv2.VideoCapture(0)
                        for i in range(10):
                                status, image = cap.read()
                else:
                        image = cv2.imread('00000001.jpg')

                self.image = cv2.undistort(image, CMATRIX, DIST, None, NCMATRIX)

                # Get various data about the image from the user
                self.get_pitch_outline()

                self.get_zone('Zone_0', 'draw LEFT Defender')
                self.get_zone('Zone_1', 'draw LEFT Attacker')
                self.get_zone('Zone_2', 'draw RIGHT Attacker')
                self.get_zone('Zone_3', 'draw RIGHT Defender')

                self.get_goal('Zone_0')
                self.get_goal('Zone_3')

                print 'Press any key to finish.'
                cv2.waitKey(0)
                cv2.destroyAllWindows()

                # Write out the data
                # self.dump('calibrations/calibrate.json', self.data)
                tools.save_croppings(pitch=self.pitch, data=self.data)
Example #2
0
    def run(self, camera=False):
        frame = cv2.namedWindow(FRAME_NAME)

        # Set callback
        cv2.setMouseCallback(FRAME_NAME, self.draw)

        if camera:
            self.image = self.camera.get_raw_frame()
        else:
            self.image = cv2.imread('00000001.jpg')

        self.image = self.camera.fix_radial_distortion(self.image)
        self.image = self.camera.fix_perspective(self.image)

        # Get various data about the image from the user
        self.get_pitch_outline()

        self.get_zone('Zone_0', 'draw LEFT Defender')
        self.get_zone('Zone_1', 'draw RIGHT Defender')
        self.get_zone('Goal_0', 'Point in the center of the left goal')
        self.get_zone('Goal_1', 'Point in the cneter of the right goal')

        #self.get_goal('Zone_0')
        #self.get_goal('Zone_3')

        print 'Press any key to finish.'
        cv2.waitKey(0)
        cv2.destroyAllWindows()

        # Write out the data
        # self.dump('calibrations/calibrate.json', self.data)
        tools.save_croppings(pitch=self.pitch, data=self.data)
Example #3
0
    def run(self, camera=False):
        frame = cv2.namedWindow(FRAME_NAME)

        # Set callback
        cv2.setMouseCallback(FRAME_NAME, self.draw)

        if camera:
            cap = cv2.VideoCapture(0)
            for i in range(10):
                status, image = cap.read()
        else:
            image = cv2.imread('00000001.jpg')

        self.image = cv2.undistort(image, CMATRIX, DIST, None, NCMATRIX)

        # Get various data about the image from the user
        self.get_pitch_outline()

        self.get_zone('Zone_0', 'draw LEFT Defender')
        self.get_zone('Zone_1', 'draw LEFT Attacker')
        self.get_zone('Zone_2', 'draw RIGHT Attacker')
        self.get_zone('Zone_3', 'draw RIGHT Defender')

        self.get_goal('Zone_0')
        self.get_goal('Zone_3')

        print 'Press any key to finish.'
        cv2.waitKey(0)
        cv2.destroyAllWindows()

        # Write out the data
        # self.dump('calibrations/calibrate.json', self.data)
        tools.save_croppings(pitch=self.pitch, data=self.data)
Example #4
0
        def run(self, camera=False):
                '''
                Run constructs a new OpenCV window, displays the frame for cropping and
                begins the routine of the program asking for user input.
                :param camera: True indicates using the camera feed, False will use a \
                        image entitled '00000001.jpg' instead.
                '''
                frame = cv2.namedWindow(FRAME_NAME)

                # Set mouse callback to the self.draw function
                cv2.setMouseCallback(FRAME_NAME, self.draw)

                # If we're using the camera, pull multiple images to ensure
                # we have a valid one. (Initial pulled images can be strange.)
                if camera:
                        cap = cv2.VideoCapture(0)
                        for i in range(10):
                                status, image = cap.read()

                # Otherwise, use plain image
                else:
                        image = cv2.imread('00000001.jpg')

                # Apply barrel distortion fix to the image
                self.image = cv2.undistort(image, CMATRIX, DIST, None, NCMATRIX)

                self.get_zone('Zone_0', 'draw LEFT Defender')
                self.get_zone('Zone_1', 'draw LEFT Attacker')
                self.get_zone('Zone_2', 'draw RIGHT Attacker')
                self.get_zone('Zone_3', 'draw RIGHT Defender')

		self.data['Zone_3'].extend(self.data['Zone_2'])
		self.data['Zone_2'].extend(self.data['Zone_1'])
		self.data['Zone_1'].extend(sorted(self.data['Zone_0'], key = lambda x : x[0], reverse = True)[:2])
		minx = min(self.data['outline'], key = lambda x: x[0])[0]
		miny = min(self.data['outline'], key = lambda x: x[1])[1]

		for k in ['Zone_0', 'Zone_1', 'Zone_2', 'Zone_3']:
			self.data[k] = [(x-minx, y-miny) for (x,y) in self.data[k]]
		for k in ['outline', 'Zone_0', 'Zone_1', 'Zone_2', 'Zone_3']:
			self.data[k] = self.sortPoly(self.data[k])

                # Calculate the goal positions for each zone
                self.get_goal('Zone_0')
                self.get_goal('Zone_3')

		self.draw_poly(self.reshape())
                print 'Press any key to finish.'
                cv2.waitKey(0)
                cv2.destroyAllWindows()

                # Write out the data
                tools.save_croppings(pitch=self.pitch, data=self.data)