国产成人精品久久免费动漫-国产成人精品天堂-国产成人精品区在线观看-国产成人精品日本-a级毛片无码免费真人-a级毛片毛片免费观看久潮喷

您的位置:首頁(yè)技術(shù)文章
文章詳情頁(yè)

python目標(biāo)檢測(cè)給圖畫框,bbox畫到圖上并保存案例

瀏覽:48日期:2022-08-02 18:44:37

我就廢話不多說(shuō)了,還是直接上代碼吧!

import osimport xml.dom.minidomimport cv2 as cv ImgPath = ’C:/Users/49691/Desktop/gangjin/gangjin_test/JPEGImages/’AnnoPath = ’C:/Users/49691/Desktop/gangjin/gangjin_test/Annotations/’ #xml文件地址save_path = ’’def draw_anchor(ImgPath,AnnoPath,save_path): imagelist = os.listdir(ImgPath) for image in imagelist: image_pre, ext = os.path.splitext(image) imgfile = ImgPath + image xmlfile = AnnoPath + image_pre + ’.xml’ # print(image) # 打開xml文檔 DOMTree = xml.dom.minidom.parse(xmlfile) # 得到文檔元素對(duì)象 collection = DOMTree.documentElement # 讀取圖片 img = cv.imread(imgfile) filenamelist = collection.getElementsByTagName('filename') filename = filenamelist[0].childNodes[0].data print(filename) # 得到標(biāo)簽名為object的信息 objectlist = collection.getElementsByTagName('object') for objects in objectlist: # 每個(gè)object中得到子標(biāo)簽名為name的信息 namelist = objects.getElementsByTagName(’name’) # 通過(guò)此語(yǔ)句得到具體的某個(gè)name的值 objectname = namelist[0].childNodes[0].data bndbox = objects.getElementsByTagName(’bndbox’) # print(bndbox) for box in bndbox:x1_list = box.getElementsByTagName(’xmin’)x1 = int(x1_list[0].childNodes[0].data)y1_list = box.getElementsByTagName(’ymin’)y1 = int(y1_list[0].childNodes[0].data)x2_list = box.getElementsByTagName(’xmax’) #注意坐標(biāo),看是否需要轉(zhuǎn)換x2 = int(x2_list[0].childNodes[0].data)y2_list = box.getElementsByTagName(’ymax’)y2 = int(y2_list[0].childNodes[0].data)cv.rectangle(img, (x1, y1), (x2, y2), (255, 255, 255), thickness=2)cv.putText(img, objectname, (x1, y1), cv.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), thickness=2)# cv.imshow(’head’, img)cv.imwrite(save_path+’/’+filename, img) #save picture

補(bǔ)充知識(shí):深度學(xué)習(xí)python之用Faster-rcnn 檢測(cè)結(jié)果(txt文件) 在原圖畫出box

使用Faster-rcnn 的test_net.py 檢測(cè)網(wǎng)絡(luò)的mAP等精度會(huì)生成一個(gè)檢測(cè)結(jié)果(txt文件),格式如下:

000004 0.972 302.8 94.5 512.0 150.0000004 0.950 348.1 166.1 512.0 242.9000004 0.875 1.0 25.7 292.6 126.3000004 0.730 1.0 138.5 488.3 230.0000004 0.699 1.0 120.9 145.5 139.9000004 0.592 54.4 227.4 431.9 343.4000004 0.588 1.0 159.8 18.8 231.6000004 0.126 1.0 247.1 342.3 270.0000004 0.120 1.0 225.4 185.7 309.3

每行分別為 名稱 檢測(cè)概率 xmin ymin xmax ymax

問(wèn)題在于每一行只顯示一個(gè)box數(shù)據(jù),每幅圖像可能包括多個(gè)box,需要判斷提取的多行數(shù)據(jù)是不是屬于同一圖片

下面使用python提取這些數(shù)據(jù),在原圖上畫出box并且保存起來(lái)

import osimport os.pathimport numpy as npimport xml.etree.ElementTree as xmlETfrom PIL import Image, ImageDrawimport cPickle as pickle txt_name = ’comp4_8a226fd7-753d-40fc-8013-f68d2a465579_det_test_ship.txt’file_path_img = ’/home/JPEGImages’save_file_path = ’/home/detect_results’source_file = open(txt_name)img_names = []for line in source_file: staff = line.split() img_name = staff[0] img_names.append(img_name)name_dict = {}for i in img_names: if img_names.count(i)>0: name_dict[i] = img_names.count(i) source_file.close()source_file = open(txt_name)for idx in name_dict: img = Image.open(os.path.join(file_path_img, idx + ’.jpg’)) draw = ImageDraw.Draw(img) for i in xrange(name_dict[idx]): line = source_file.readline() staff = line.split() score = staff[1] box = staff[2:6] draw.rectangle([int(np.round(float(box[0]))), int(np.round(float(box[1]))), int(np.round(float(box[2]))), int(np.round(float(box[3])))], outline=(255, 0, 0)) img.save(os.path.join(save_file_path, idx + ’.jpg’)) source_file.close()

運(yùn)行完即可在保存文件夾中得到效果圖。

以上這篇python目標(biāo)檢測(cè)給圖畫框,bbox畫到圖上并保存案例就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持好吧啦網(wǎng)。

標(biāo)簽: Python 編程
主站蜘蛛池模板: 亚洲综合欧美综合 | a级毛片毛片免费观看永久 a级毛片毛片免费很很综合 | 亚洲欧美久久 | 久久99国产乱子伦精品免费 | 日本三级s级在线播放 | 欧美成人午夜毛片免费影院 | 日韩一级大毛片欧美一级 | 久久91精品国产一区二区 | 在线观看人成网站深夜免费 | 日本免费成人网 | 久久免费国产精品一区二区 | 成人做爰全过程免费看网站 | 国外精品视频在线观看免费 | 美国欧美一级毛片 | 久久99久久精品国产99热 | 国产黄色片网站 | 成年人在线网站 | 久久精品免费全国观看国产 | 午夜免费片在线观看不卡 | 怡红院免费的全部视频国产a | 欧美成人精品高清在线播放 | 在线看片日韩 | 国产激情一区二区三区 | 欧美人在线 | 自拍视频在线观看视频精品 | 日韩国产免费 | 精品一区二区三区在线成人 | 欧美一级欧美三级在线 | 日本久久久久一级毛片 | 国产成人午夜性a一级毛片 国产成人午夜性视频影院 国产成人香蕉久久久久 | 久色成人 | 久久精品亚洲一级毛片 | 久久成人在线 | 久久精品3 | 日韩中文字幕视频 | 中文字幕一二三区乱码老 | 国产欧美一区二区三区沐欲 | 欧美精品日日鲁夜夜添 | 免费成年人视频网站 | 黄视频在线免费 | 黄色成人免费网站 |