def load_train_mat_label(trainpath, files): res_list = [] for file in files: names = file.split('.') _ord = int(names[-2]) label = chr(_ord) img = Image.open('{}/{}'.format(trainpath, file)) mat = img2mat(img) res_list.append((mat, label)) return res_list
def load_train_mat_label(trainpath,files): res_list = [] for file in files: names = file.split('.') _ord = int(names[-2]) label = chr(_ord) img = Image.open('{}/{}'.format(trainpath,file)) mat = img2mat(img) res_list.append((mat,label)) return res_list
def split_img(bin_img,col_rate=0.7,row_rate=0.4,count=None): mat = img2mat(bin_img) splits = [] res_list = split_mat(mat,col_rate=col_rate,row_rate=row_rate,count=count) for res in res_list: tp,_ = res r1,r2,c1,c2 = tp _img = bin_img.crop((c1,r1,c2,r2)) splits.append((tp,_img)) return splits
def split_img(bimg, **kwargs): if kwargs == {}: kwargs = split_kwargs mat = img2mat(bimg) splits = [] results = split_mat(mat, **kwargs) for result in results: indexes, _ = result r1, r2, c1, c2 = indexes img = bimg.crop((c1, r1, c2, r2)) splits.append((indexes, img)) return splits
def split_img(bin_img, col_rate=0.7, row_rate=0.4, count=None): mat = img2mat(bin_img) splits = [] res_list = split_mat(mat, col_rate=col_rate, row_rate=row_rate, count=count) for res in res_list: tp, _ = res r1, r2, c1, c2 = tp _img = bin_img.crop((c1, r1, c2, r2)) splits.append((tp, _img)) return splits
def rename(splitpath, trainpath, chars, cnt=3): _chars = {} for char in chars: print(char, cnt) _chars[char] = cnt files = os.listdir(splitpath) for file in files: img = Image.open('{}/{}'.format(splitpath, file)) #img = img mat = img2mat(img) show_pixel(mat) while True: char = raw_input( 'enter the char (if want pass just press `Enter`):') if char == '': break if char not in _chars: print('char `{}` not in the chars,please check'.format(char)) elif _chars[char] <= 0: break else: _chars[char] -= 1 i = _chars[char] fname = '.'.join([file, str(i), char, str(ord(char)), 'bmp']) img.save('{}/{}'.format(trainpath, fname)) break #check miss print('still miss chars : ', end='') miss = False for char in _chars: need = _chars[char] if need > 0: miss = True print('`{}`#{},'.format(char, need), end='') print('') if miss == False: break
def rename(splitpath, trainpath, count, **kwargs): chars = get_chars(**kwargs) chars_map = {} for char in chars: chars_map[char] = count files = os.listdir(splitpath) files = filter(lambda x: not x.startswith('.'), files) for file in files: img = Image.open('{}/{}'.format(splitpath, file)) mat = img2mat(img) show_pixel(mat) while True: char = raw_input('enter the char (if want pass just press `Enter`):') if char == '': break if char not in chars_map: click.echo('!!! char `{}` not in the chars,please check'.format(char)) continue elif chars_map[char] > 0: chars_map[char] -= 1 i = chars_map[char] fname = '.'.join([file, str(i), char, str(ord(char)), 'bmp']) img.save('{}/{}'.format(trainpath, fname)) break miss = filter(lambda x: x[1] > 0, chars_map.iteritems()) # check miss if not miss: break miss = map(lambda x: '`{0}`#{1}'.format(*x), miss) click.echo('still miss chars : {}'.format(','.join(miss)))
def rename(splitpath,trainpath,chars,cnt=3): _chars = {} for char in chars: print(char,cnt) _chars[char] = cnt files = os.listdir(splitpath) for file in files: img = Image.open('{}/{}'.format(splitpath,file)) #img = img mat = img2mat(img) show_pixel(mat) while True: char = raw_input('enter the char (if want pass just press `Enter`):') if char == '': break if char not in _chars: print('char `{}` not in the chars,please check'.format(char)) elif _chars[char] <= 0: break else: _chars[char] -= 1 i = _chars[char] fname = '.'.join([file,str(i),char,str(ord(char)),'bmp']) img.save('{}/{}'.format(trainpath,fname)) break #check miss print('still miss chars : ',end='') miss = False for char in _chars: need = _chars[char] if need >0: miss = True print('`{}`#{},'.format(char,need),end='') print('') if miss == False: break
def read_raw_img(raw_img): bin_img = chg_img(raw_img) mat = img2mat(bin_img) #print_mat(mat,replace=lambda x:' ' if x<200 else 'M') result = read_mat(mat) return result
return result def read_mat(mat): res_list = split_mat(mat,**split_kwargs) result = [] for res in res_list: tp,_mat = res #print_mat(_mat,replace=lambda x:' ' if x<200 else 'M') label,sim = classfiy(_mat) result.append((label,sim)) return result if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('-c','--classfiy',action='store',help='give a split image ') parser.add_argument('-r','--readraw',action='store',help='give a raw image') args = parser.parse_args() if args.classfiy != None: img = Image.open(args.classfiy) mat = img2mat(img) result = classfiy(mat) print(result) if args.readraw != None: raw_img = Image.open(args.readraw) labels = read_raw_img(raw_img) print(labels)
def read_img(img): bimg = bin_img(img) mat = img2mat(bimg) result = read_mat(mat) return result
def read_raw_img(raw_img): bin_img = chg_img(raw_img) mat = img2mat(bin_img) #print_mat(mat,replace=lambda x:' ' if x<200 else 'M') result = read_mat(mat) return result
for res in res_list: tp, _mat = res #print_mat(_mat,replace=lambda x:' ' if x<200 else 'M') label, sim = classfiy(_mat) result.append((label, sim)) return result if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('-c', '--classfiy', action='store', help='give a split image ') parser.add_argument('-r', '--readraw', action='store', help='give a raw image') args = parser.parse_args() if args.classfiy != None: img = Image.open(args.classfiy) mat = img2mat(img) result = classfiy(mat) print(result) if args.readraw != None: raw_img = Image.open(args.readraw) labels = read_raw_img(raw_img) print(labels)