1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
| import cv2 import os import numpy as np from PIL import Image
def mouse_img_cod(event, cod_x, cod_y, flags, param): if event == cv2.EVENT_LBUTTONDOWN: xy = "%d,%d" % (cod_x,cod_y) cv2.circle(img, (cod_x,cod_y), 1, (255, 0, 0), thickness = -1) cv2.putText(img, xy, (cod_x,cod_y), cv2.FONT_HERSHEY_PLAIN,1.0, (0,0,0), thickness = 1) cv2.imshow("image", img) pro_x.append(cod_x) pro_y.append(cod_y)
path = r"E:\doc\traindatasets\img_rec" for name in os.listdir(path): img_path = os.path.join(path, name) img = cv2.imread(img_path)[:,:,:3] pro_x = [] pro_y = [] cv2.namedWindow('image', cv2.WINDOW_AUTOSIZE) cv2.setMouseCallback("image", mouse_img_cod) cv2.imshow('image',img) cv2.waitKey(0) cv2.destroyAllWindows() print(pro_y[0], pro_x[0]) img = cv2.imread(img_path)[:,:,:3] B,G,R = img[pro_y[0], pro_x[0], :] print("file: {} RGB: {} {} {}".format(name, R, G, B)) file = open("color3.txt", "a+") file.write("file: {} RGB: {} {} {}\n".format(name, R, G, B)) file.close()
|