Example-production of rice segmentation data set

Practical example-Production of rice segmentation data set-Production of rice segmentation data set Real CasesWater rice fraction cut the number of data sets prepared for

  • 1. Read the original image and divide it into 3 x 5 blocks
  • 2. Save separately
  • 3. Manually make data sets
img_raw = cv2.imread('r1.png')
# img_raw.shape
# begin = 0
# gap = 400

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

Insert picture description here
Insert picture description here
Insert picture description here

Guess you like

Origin blog.csdn.net/qq_41375318/article/details/108521338