cnn_db_loader.py 文件源码

def get_mean_image_path(self):
        if not os.path.exists(self.outputfolder+'/'+self.mean_image_filename):
            print('no mean image found. Creating...')
            isomap_size = cv2.imread(self.examples_train[0].images[0], cv2.IMREAD_COLOR).shape[0]
            mean = np.zeros([isomap_size, isomap_size, 3], dtype='float32')

            for example in self.examples_train:#
                try:
                    mean+=cv2.imread(example.images[0],cv2.IMREAD_COLOR).astype(dtype='float32')/len(self.examples_train)
                except:
                    e = sys.exc_info()[0]
                    print (str(e))
                    print ('image', example.images[0])
                    exit(0)
            #mean/=len(self.images_train)
            mean_uint8 = mean.astype(dtype='uint8')
            cv2.imwrite(self.outputfolder+'/'+self.mean_image_filename, mean_uint8)
        return self.outputfolder+'/'+self.mean_image_filename