generater.py 文件源码

def build_identify(self, writer):
        X_list = []
        dict_map = {}

        train_writer_path = '{:s}/{:s}'.format(conf.train_path, writer)
        files = os.listdir(train_writer_path)
        for file in files:
            tmp_path = '{:s}/{:s}'.format(train_writer_path, file)
            img_data = self.read(tmp_path)
            if img_data is None: continue
            X_list.append(img_data)

            letter = self.predict(np.array(img_data))[0]
            if letter not in dict_map:
                dict_map[letter] = 0
            else:
                dict_map[letter] += 1
            print('=> rename: {:s} => {:s}_{:d}.jpg'.format(file, letter, dict_map[letter]))
            os.rename(tmp_path, '{:s}/{:s}_{:d}.jpg'.format(train_writer_path, letter, dict_map[letter]))
        return self