Art_class.py 文件源码

def get_hsv(self, plot=False):
        """
        Extract HSV values for each image. Creates bins for the HSV vectors.
        Also finds the peaks in the HSV histograms

        Input:  None
        Output: None
        """
        self.hsv_image = color.rgb2hsv(self.image)
        self.hue_bins, self.avg_hue, self.hue_var = self.create_hist_vector(self.hsv_image, 0, 48, (0.0, 1))
        self.sat_bins, self.avg_sat, self.sat_var = self.create_hist_vector(self.hsv_image, 1, 32, (0.0, 1))
        self.val_bins, self.avg_val, self.val_var = self.create_hist_vector(self.hsv_image, 2, 32, (0.0, 1))
        # get the peaks
        self.hue_peaks = self.get_peaks(self.hue_bins, 0.5, 5)
        self.val_peaks = self.get_peaks(self.val_bins, 0.4, 5)
        self.sat_peaks = self.get_peaks(self.sat_bins, 0.4, 5)
        if plot is True:
            viz.plot_hsv(self.hsv_image)