潍坊程序设计网站建设公司,公众号制作开发公司,seo公司哪家好用,c#网站开发框架Git链接
安装按照上面的说明#xff0c;说下使用。
把tools下面的test做了一点修改#xff0c;可以读取一张图片#xff0c;把里面的单个字符都检测和识别出来。
然后绘制到屏幕上。 import torch
from charnet.modeling.model import CharNet
import cv2, os
import num…Git链接
安装按照上面的说明说下使用。
把tools下面的test做了一点修改可以读取一张图片把里面的单个字符都检测和识别出来。
然后绘制到屏幕上。 import torch
from charnet.modeling.model import CharNet
import cv2, os
import numpy as np
import argparse
from charnet.config import cfgdef loadDict():fn_dicttools\char_dict.txtwith open(fn_dict, r) as file:lines file.readlines()# 去除每行末尾的换行符lines [line.strip() for line in lines]dict_char{}for line in lines:lineline.replace(\x1f,)num_linelen(line)aline[0]indexline[1:]indexint(index)dict_char[index]areturn dict_chardef resize(im, size):h, w, _ im.shapescale max(h, w) / float(size)image_resize_height int(round(h / scale / cfg.SIZE_DIVISIBILITY) * cfg.SIZE_DIVISIBILITY)image_resize_width int(round(w / scale / cfg.SIZE_DIVISIBILITY) * cfg.SIZE_DIVISIBILITY)scale_h float(h) / image_resize_heightscale_w float(w) / image_resize_widthim cv2.resize(im, (image_resize_width, image_resize_height), interpolationcv2.INTER_LINEAR)return im, scale_w, scale_h, w, hif __name__ __main__:dict_charloadDict()parser argparse.ArgumentParser(descriptionTest)fn_confrconfigs\icdar2015_hourglass88.yamlfn_weightrconfigs\icdar2015_hourglass88.pthargs parser.parse_args()cfg.merge_from_file(fn_conf)cfg.freeze()charnet CharNet()charnet.load_state_dict(torch.load(fn_weight))charnet.eval()charnet.cuda()im_filerdata\2.jpgim_original cv2.imread(im_file)im, scale_w, scale_h, original_w, original_h resize(im_original, sizecfg.INPUT_SIZE)with torch.no_grad():char_bboxes, char_scores, word_instances charnet(im, scale_w, scale_h, original_w, original_h)for ic,box in enumerate(char_bboxes):print(box)scorechar_scores[ic]max_index np.argmax(score)labeldict_char[max_index]points np.array(box[0:8]).reshape(-1, 2).astype(np.int32)cv2.polylines(im_original, [points], isClosedTrue, color(0, 0, 255), thickness1)font cv2.FONT_HERSHEY_SIMPLEXcv2.putText(im_original, label, (points[0][0],points[0][1]), font, 1, (0, 255, 0), 1)cv2.imshow(img,im_original)cv2.waitKey(0)
