代碼鏈接:

功能：

制作圖片像素點(diǎn)數(shù)據(jù)集。

打開文件夾中的圖片，通過鼠標(biāo)點(diǎn)擊圖片獲取所在鼠標(biāo)點(diǎn)的像素值，并把打的標(biāo)簽導(dǎo)出為txt。

導(dǎo)入必要的函數(shù)庫

import math import os import cv2 as cv import numpy as np

定義一些全局變量后面會(huì)用到。patchsize表示采集的像素patch大小。

mode = True testflag = False idex = 0 patchsize = 9 # 對(duì)應(yīng)像素塊大小 # 讀取圖片 dir = './pic' # 圖片存放的目錄

定義讀取圖片的函數(shù)，方便后面循環(huán)調(diào)用文件夾內(nèi)的圖片。讀取后的圖片是narray的格式。

def readImage(dir, idex):"""讀取圖片"""filelist = os.listdir(dir)imgname = os.path.join(dir, filelist[idex])img1 = cv.imread(imgname) # 這里必須用cv庫里面的imread，否則格式不對(duì)會(huì)報(bào)錯(cuò)return img1

定義bgr_patch函數(shù)獲得像素所處patch的像素平均值

def bgr_patch(img, y, x):"""輸入圖片img，輸入對(duì)應(yīng)的坐標(biāo)y，x，輸出這個(gè)坐標(biāo)周圍一個(gè)patch的像素平均值"""global patchsizen = math.floor(patchsize / 2)bgr = img[y - n:y + (n + 1), x - n:x + (n + 1)] # 選取 5*5=25 的區(qū)域size = bgr.shape[0]sum1 = np.array([0, 0, 0])for i in range(size):for j in range(size):sum1 = sum1 + bgr[i, j]ave = sum1 / (size * size)ave = np.array([round(ave[0]), round(ave[1]), round(ave[2])])print(f"bgr test is {ave}")return ave

定義一個(gè)窗口函數(shù)，作為setMouseCallback的回調(diào)函數(shù)。在窗口函數(shù)中，通過鼠標(biāo)的不同操作和鍵盤的操作實(shí)現(xiàn)不同的功能。

• x，y:鼠標(biāo)所在的位置

# 定義一個(gè)窗口 def draw_(event,x, y,flag,param):global idex, dir, mode, data, img1, testflag, patchsizen = math.floor(patchsize / 2)filelist = os.listdir(dir)if event == cv.EVENT_LBUTTONDOWN:"""左鍵單擊打標(biāo)簽，按e打環(huán)境標(biāo)簽，按r打道路標(biāo)簽，按t切換測(cè)試集，中間滾動(dòng)切換圖片，按c退出"""xy = "%d,%d" % (x, y)# bgr = img1[y, x] # 按照BGR排列,x和y是相反的# print(f"bgr origin is {bgr}")bgr = bgr_patch(img1, y, x) # 使用patchif mode:"""道路"""rgb = [bgr[2], bgr[1], bgr[0], y, x, 1] # 轉(zhuǎn)換成RGB排列y_size = img1.shape[0]x_size = img1.shape[1]rgb = [bgr[2] / 255, bgr[1] / 255, bgr[0] / 255,x / x_size, y / y_size, 1] # 歸一化 數(shù)值取到0——1之間print(f"rgb guiyi is {rgb}")cv.circle(img1, (x, y), n, (255, 0, 0), thickness=-1)# cv.putText(img1,xy,(x,y),fontFace=cv.FONT_HERSHEY_PLAIN# fontScale=2.0,color=(0,0,0),thickness= 1) # 顯示標(biāo)簽值data.append(rgb)else:"""非道路"""rgb = [bgr[2], bgr[1], bgr[0], y, x, 0] # 轉(zhuǎn)換成RGB排列y_size = img1.shape[0]x_size = img1.shape[1]rgb = [bgr[2] / 255, bgr[1] / 255, bgr[0] / 255, x / x_size,y / y_size, 0] # 歸一化 數(shù)值取到0——1之間print(f"rgb guiyi is {rgb}")cv.circle(img1, (x, y), n, (0, 0, 255), thickness=-1)# cv.putText(img1,xy,(x,y),fontFace=cv.FONT_HERSHEY_PLAIN,# fontScale=2.0,color=(0,0,0),thickness= 1) # 顯示標(biāo)簽值data.append(rgb)if event == cv.EVENT_MOUSEWHEEL:"""滾動(dòng)觸發(fā)圖片切換"""if idex != len(filelist) - 2:idex = idex + 1else:idex = 0img1 = readImage(dir, idex)if event == cv.EVENT_MBUTTONDOWN:"""單擊鼠標(biāo)中間，存儲(chǔ)數(shù)據(jù)"""if testflag:file = open('data_test.txt', mode='w+')file.writelines(str(data))file.close()else:file = open('data_train.txt', mode='w+')file.writelines(str(data))file.close()

創(chuàng)建opencv窗口,命名為image，設(shè)置鼠標(biāo)響應(yīng)函數(shù)

cv.namedWindow('image') cv.setMouseCallback('image', draw_)

主函數(shù)寫在while循環(huán)里面。

# 創(chuàng)建一個(gè)列表存儲(chǔ)標(biāo)簽和RGB值 data = [] label = [] img1 = readImage(dir, idex)while 1:cv.imshow('image', img1)k = cv.waitKey(1) & 0xFFif k == ord('r'):mode = Trueelif k == ord('e'):mode = Falseelif k == ord('t'):testflag = ~testflagprint(f"[testflag] is :{testflag}")elif k == ord('c'):break cv.destroyAllWindows()

總結(jié)

以上是生活随笔為你收集整理的简易图片打像素标签工具的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。