实例-水稻分割数据集制作

$$实例-水稻分割数据集制作$$

• 1.读取原图分成3 x 5块
• 2.分别保存
• 3.手动制作数据集

print("startX:",startX,"startY:", startY,"cropWidth:",cropWidth,"cropHeight:",cropHeight)
cropImg = img[startY:startY+cropHeight,startX:startX+cropWidth]

img_raw = cv2.imread('r1.png')
# img_raw.shape
# begin = 0
# gap = 400

#[0:400,400:800] 
# 0:400表示 y纵坐标
# 400:800 表示x 横坐标 
block1_1 = img_raw[0:400,0:400]
block1_2 = img_raw[0:400,400:800]
block1_3 = img_raw[0:400,800:1200]
block1_4 = img_raw[0:400,1200:1600]
block1_5 = img_raw[0:400,1600:2000]
# cv2.imshow('block1_1', block1_1)
# cv2.imshow('block1_2', block1_2)
# cv2.imshow('block1_3', block1_3)
# cv2.imshow('block1_4', block1_4)
# cv2.imshow('block1_5', block1_5)


block2_1 = img_raw[400:800,0:400]
block2_2 = img_raw[400:800,400:800]
block2_3 = img_raw[400:800,800:1200]
block2_4 = img_raw[400:800,1200:1600]
block2_5 = img_raw[400:800,1600:2000]
# cv2.imshow('block2_1', block2_1)
# cv2.imshow('block2_2', block2_2)
# cv2.imshow('block2_3', block2_3)
# cv2.imshow('block2_4', block2_4)
# cv2.imshow('block2_5', block2_5)

block3_1 = img_raw[800:1200,0:400]
block3_2 = img_raw[800:1200,400:800]
block3_3 = img_raw[800:1200,800:1200]
block3_4 = img_raw[800:1200,1200:1600]
block3_5 = img_raw[800:1200,1600:2000]
# cv2.imshow('block3_1', block3_1)
# cv2.imshow('block3_2', block3_2)
# cv2.imshow('block3_3', block3_3)
# cv2.imshow('block3_4', block3_4)
# cv2.imshow('block3_5', block3_5)

cv2.imwrite(r"./dataset/image/1_block1_1.png", block1_1)
cv2.imwrite(r"./dataset/image/1_block1_2.png", block1_2)
cv2.imwrite(r"./dataset/image/1_block1_3.png", block1_3)
cv2.imwrite(r"./dataset/image/1_block1_4.png", block1_4)
cv2.imwrite(r"./dataset/image/1_block1_5.png", block1_5)


cv2.imwrite(r"./dataset/image/1_block2_1.png", block2_1)
cv2.imwrite(r"./dataset/image/1_block2_2.png", block2_2)
cv2.imwrite(r"./dataset/image/1_block2_3.png", block2_3)
cv2.imwrite(r"./dataset/image/1_block2_4.png", block2_4)
cv2.imwrite(r"./dataset/image/1_block2_5.png", block2_5)

cv2.imwrite(r"./dataset/image/1_block3_1.png", block3_1)
cv2.imwrite(r"./dataset/image/1_block3_2.png", block3_2)
cv2.imwrite(r"./dataset/image/1_block3_3.png", block3_3)
cv2.imwrite(r"./dataset/image/1_block3_4.png", block3_4)
cv2.imwrite(r"./dataset/image/1_block3_5.png", block3_5)


# cv2.waitKey(0)
# cv2.destroyAllWindows()

img_raw = cv2.imread('r2.png')
img_raw.shape

block1_1 = img_raw[0:400,0:400]
block1_2 = img_raw[0:400,400:800]
block1_3 = img_raw[0:400,800:1200]
block1_4 = img_raw[0:400,1200:1600]
block1_5 = img_raw[0:400,1600:2000]
# cv2.imshow('block1_1', block1_1)
# cv2.imshow('block1_2', block1_2)
# cv2.imshow('block1_3', block1_3)
# cv2.imshow('block1_4', block1_4)
# cv2.imshow('block1_5', block1_5)


block2_1 = img_raw[400:800,0:400]
block2_2 = img_raw[400:800,400:800]
block2_3 = img_raw[400:800,800:1200]
block2_4 = img_raw[400:800,1200:1600]
block2_5 = img_raw[400:800,1600:2000]
# cv2.imshow('block2_1', block2_1)
# cv2.imshow('block2_2', block2_2)
# cv2.imshow('block2_3', block2_3)
# cv2.imshow('block2_4', block2_4)
# cv2.imshow('block2_5', block2_5)

block3_1 = img_raw[800:1200,0:400]
block3_2 = img_raw[800:1200,400:800]
block3_3 = img_raw[800:1200,800:1200]
block3_4 = img_raw[800:1200,1200:1600]
block3_5 = img_raw[800:1200,1600:2000]
# cv2.imshow('block3_1', block3_1)
# cv2.imshow('block3_2', block3_2)
# cv2.imshow('block3_3', block3_3)
# cv2.imshow('block3_4', block3_4)
# cv2.imshow('block3_5', block3_5)

cv2.imwrite(r"./dataset/image/2_block1_1.png", block1_1)
cv2.imwrite(r"./dataset/image/2_block1_2.png", block1_2)
cv2.imwrite(r"./dataset/image/2_block1_3.png", block1_3)
cv2.imwrite(r"./dataset/image/2_block1_4.png", block1_4)
cv2.imwrite(r"./dataset/image/2_block1_5.png", block1_5)


cv2.imwrite(r"./dataset/image/2_block2_1.png", block2_1)
cv2.imwrite(r"./dataset/image/2_block2_2.png", block2_2)
cv2.imwrite(r"./dataset/image/2_block2_3.png", block2_3)
cv2.imwrite(r"./dataset/image/2_block2_4.png", block2_4)
cv2.imwrite(r"./dataset/image/2_block2_5.png", block2_5)

cv2.imwrite(r"./dataset/image/2_block3_1.png", block3_1)
cv2.imwrite(r"./dataset/image/2_block3_2.png", block3_2)
cv2.imwrite(r"./dataset/image/2_block3_3.png", block3_3)
cv2.imwrite(r"./dataset/image/2_block3_4.png", block3_4)
cv2.imwrite(r"./dataset/image/2_block3_5.png", block3_5)

img_raw = cv2.imread('r3.png')
img_raw.shape

block1_1 = img_raw[0:400,0:400]
block1_2 = img_raw[0:400,400:800]
block1_3 = img_raw[0:400,800:1200]
block1_4 = img_raw[0:400,1200:1600]
block1_5 = img_raw[0:400,1600:2000]
# cv2.imshow('block1_1', block1_1)
# cv2.imshow('block1_2', block1_2)
# cv2.imshow('block1_3', block1_3)
# cv2.imshow('block1_4', block1_4)
# cv2.imshow('block1_5', block1_5)


block2_1 = img_raw[400:800,0:400]
block2_2 = img_raw[400:800,400:800]
block2_3 = img_raw[400:800,800:1200]
block2_4 = img_raw[400:800,1200:1600]
block2_5 = img_raw[400:800,1600:2000]
# cv2.imshow('block2_1', block2_1)
# cv2.imshow('block2_2', block2_2)
# cv2.imshow('block2_3', block2_3)
# cv2.imshow('block2_4', block2_4)
# cv2.imshow('block2_5', block2_5)

block3_1 = img_raw[800:1200,0:400]
block3_2 = img_raw[800:1200,400:800]
block3_3 = img_raw[800:1200,800:1200]
block3_4 = img_raw[800:1200,1200:1600]
block3_5 = img_raw[800:1200,1600:2000]
# cv2.imshow('block3_1', block3_1)
# cv2.imshow('block3_2', block3_2)
# cv2.imshow('block3_3', block3_3)
# cv2.imshow('block3_4', block3_4)
# cv2.imshow('block3_5', block3_5)

cv2.imwrite(r"./dataset/image/3_block1_1.png", block1_1)
cv2.imwrite(r"./dataset/image/3_block1_2.png", block1_2)
cv2.imwrite(r"./dataset/image/3_block1_3.png", block1_3)
cv2.imwrite(r"./dataset/image/3_block1_4.png", block1_4)
cv2.imwrite(r"./dataset/image/3_block1_5.png", block1_5)


cv2.imwrite(r"./dataset/image/3_block2_1.png", block2_1)
cv2.imwrite(r"./dataset/image/3_block2_2.png", block2_2)
cv2.imwrite(r"./dataset/image/3_block2_3.png", block2_3)
cv2.imwrite(r"./dataset/image/3_block2_4.png", block2_4)
cv2.imwrite(r"./dataset/image/3_block2_5.png", block2_5)

cv2.imwrite(r"./dataset/image/3_block3_1.png", block3_1)
cv2.imwrite(r"./dataset/image/3_block3_2.png", block3_2)
cv2.imwrite(r"./dataset/image/3_block3_3.png", block3_3)
cv2.imwrite(r"./dataset/image/3_block3_4.png", block3_4)
cv2.imwrite(r"./dataset/image/3_block3_5.png", block3_5)

img_raw = cv2.imread('r4.png')
img_raw.shape

block1_1 = img_raw[0:400,0:400]
block1_2 = img_raw[0:400,400:800]
block1_3 = img_raw[0:400,800:1200]
block1_4 = img_raw[0:400,1200:1600]
block1_5 = img_raw[0:400,1600:2000]
# cv2.imshow('block1_1', block1_1)
# cv2.imshow('block1_2', block1_2)
# cv2.imshow('block1_3', block1_3)
# cv2.imshow('block1_4', block1_4)
# cv2.imshow('block1_5', block1_5)


block2_1 = img_raw[400:800,0:400]
block2_2 = img_raw[400:800,400:800]
block2_3 = img_raw[400:800,800:1200]
block2_4 = img_raw[400:800,1200:1600]
block2_5 = img_raw[400:800,1600:2000]
# cv2.imshow('block2_1', block2_1)
# cv2.imshow('block2_2', block2_2)
# cv2.imshow('block2_3', block2_3)
# cv2.imshow('block2_4', block2_4)
# cv2.imshow('block2_5', block2_5)

block3_1 = img_raw[800:1200,0:400]
block3_2 = img_raw[800:1200,400:800]
block3_3 = img_raw[800:1200,800:1200]
block3_4 = img_raw[800:1200,1200:1600]
block3_5 = img_raw[800:1200,1600:2000]
# cv2.imshow('block3_1', block3_1)
# cv2.imshow('block3_2', block3_2)
# cv2.imshow('block3_3', block3_3)
# cv2.imshow('block3_4', block3_4)
# cv2.imshow('block3_5', block3_5)

cv2.imwrite(r"./dataset/image/4_block1_1.png", block1_1)
cv2.imwrite(r"./dataset/image/4_block1_2.png", block1_2)
cv2.imwrite(r"./dataset/image/4_block1_3.png", block1_3)
cv2.imwrite(r"./dataset/image/4_block1_4.png", block1_4)
cv2.imwrite(r"./dataset/image/4_block1_5.png", block1_5)


cv2.imwrite(r"./dataset/image/4_block2_1.png", block2_1)
cv2.imwrite(r"./dataset/image/4_block2_2.png", block2_2)
cv2.imwrite(r"./dataset/image/4_block2_3.png", block2_3)
cv2.imwrite(r"./dataset/image/4_block2_4.png", block2_4)
cv2.imwrite(r"./dataset/image/4_block2_5.png", block2_5)

cv2.imwrite(r"./dataset/image/4_block3_1.png", block3_1)
cv2.imwrite(r"./dataset/image/4_block3_2.png", block3_2)
cv2.imwrite(r"./dataset/image/4_block3_3.png", block3_3)
cv2.imwrite(r"./dataset/image/4_block3_4.png", block3_4)
cv2.imwrite(r"./dataset/image/4_block3_5.png", block3_5)

labelme