定制网站开发食道里感觉有东西堵,免费下载百度,阅读网站建设规划书,网上商城网站设计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)
