2022年 11月 16日

Python：图像标注工具
  1. 作用：在图像上左键点击，显示红点，并记录点的像素坐标；右键时保存带点的图像并把点击的坐标和坐标总数写入.npy文件。
  2. 目录./imgs存放图像，./outputs存放标注结果。
  3. 类IMG作为传给鼠标回调函数的参数，它的成员data代表图像，name代表标注结果的路径，click_count代表一幅图像上的点数，points代表一幅图像上点的坐标。
  1. #encoding=utf-8
  2. import cv2 as cv
  3. import os
  4. import numpy as np
  5.  
  6.  
  7. class IMG:
  8.     data = None
  9.     name = None
  10.     click_count = 0
  11.     points = []
  12.  
  13.  
  14. def mouse_event(event, x, y, flags, item):
  15.     if event == cv.EVENT_LBUTTONDOWN:
  16.         cv.circle(item.data, (x, y), 3, (0, 0, 255), 2)
  17.         item.click_count += 1
  18.         item.points.append((x, y))
  19.         cv.imshow(item.name.split('/')[1], item.data)
  20.         cv.waitKey(0)
  21.     elif event == cv.EVENT_RBUTTONDOWN:
  22.         cv.imwrite(item.name + ".jpg", item.data)
  23.         item.points.append(item.click_count)
  24.         np.save(item.name + '_points_' + str(item.click_count) + '.npy', item.points)
  25.         cv.destroyWindow(item.name.split('/')[1])
  26.  
  27.  
  28. if __name__ == '__main__':
  29.     imgs_dir = 'imgs/'
  30.     output_dir = 'outputs/'
  31.     if not os.path.exists(output_dir):
  32.         os.mkdir(output_dir)
  33.  
  34.     img_names = os.listdir(imgs_dir)
  35.     for img_name in img_names:
  36.         img = cv.imread(imgs_dir + img_name)
  37.         item = IMG()
  38.         item.points = []
  39.         item.data = img
  40.         item.name = output_dir + img_name.split('.')[0]
  41.         
  42.         cv.namedWindow(img_name.split('.')[0])
  43.         cv.setMouseCallback(img_name.split('.')[0], mouse_event, item)
  44.         cv.imshow(img_name.split('.')[0], img)
  45.         cv.waitKey(0)