给定分割图(图上有多个同类物体,每个物体灰度值不同),给每个物体画框:
from PIL import Image
import numpy as np
from PIL import Image, ImageDraw
seg = Image.open("1.png")
segnp = np.array(seg)
# rows =[]
# columns = []
# rmins = []
# rmaxs = []
# cmins = []
# cmaxs = []
# for i in range(1,256):
# if i in segnp:
# mask = (segnp == i)
# for r in range(0,474):
# if True in mask[r,:]:
# rows.append(r)
# for c in range(0,506):
# if True in mask[:,c]:
# columns.append(c)
# rmins.append(rows[0])
# rmaxs.append(rows[-1])
# cmins.append(columns[0])
# cmaxs.append(columns[-1])
# rows.clear()
# columns.clear()
for i in range(1,256):
if i in segnp:
mask = (segnp == i)
rows = np.any(mask, axis=1)
cols = np.any(mask, axis=0)
rmin = np.where(r