utils.py 448 B

12345678910111213
  1. import numpy as np
  2. import cv2
  3. def draw_grid(img: np.ndarray, grid_w: int, grid_h:int):
  4. """划格网"""
  5. img_h, img_w, _ = img.shape
  6. # 绘制横向网格线
  7. for i in range((img_h // grid_h)+1):
  8. cv2.line(img, (0, i*grid_h), (img_w, i*grid_h), (0, 255, 0), 2)
  9. # 绘制纵向网格线
  10. for i in range((img_w // grid_w)+1):
  11. cv2.line(img, (i*grid_w, 0), (i*grid_w, img_h), (0, 255, 0), 2)
  12. return img