seg_patch.py 文件源码

def generate(self,img_root,label_root,save_root):
        files = os.listdir(img_root)
        for file in files:
            file_path = os.path.join(img_root,file)
            label_path = os.path.join(label_root,os.path.splitext(file)[0]+'_label'+self.lab_ext)


            img = cv2.imread(file_path)
            if img is None:
                print 'please check img file path'
                exit()
            label = cv2.imread(label_path,cv2.CV_LOAD_IMAGE_UNCHANGED)
            if label is None:
                print 'please check label file ext'
                exit()
            self._generate_patches(img,label,save_root,file)