python opencv

pythonのopencvで画像を読み込む際に

python
1import numpy as np
2import cv2
3import pickle
4
5
6#関数定義
7def cannyprocess(img,data_1,count,data_2):
8    edges = cv2.Canny(img,100,200)
9    cv2.imwrite("canny\" + data_1 + '000' + count + data_2,edges)#エッジ画像保存
10    can_img = cv2.imread("canny\" + data_1 + '000' + count + data_2)#エッジ画像の読み込み
11    return can_img
12
13def contourprocess(img):
14    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
15    image, contours, hierarchy = cv2.findContours(gray, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
16    return contours
17
18def maskgenerate(img,data_1,count,data_2):
19    back = np.zeros_like(img)
20    cont = contourprocess(cannyprocess(img,data_1,count,data_2))
21    mask = cv2.drawContours(back, cont, -1, color=(255, 255, 255), thickness=-1)
22    cv2.imwrite("mask\" + data_1 + '000' + count + data_2,mask)
23    mask = cv2.imread("mask\" + data_1 + '000' + count + data_2,0)
24    return mask
25
26def mask_and_alpha(base_img, mask_img, data_1, count, data_2):
27    bgr_img = cv2.split(base_img)
28    clip = cv2.merge(bgr_img + [mask_img])
29    cv2.imwrite("clip\" + data_1 + '000' + count + data_2,clip)
30    return clip
31#関数定義
32
33#切り取り用
34#img = cv2.imread("base\2640163.jpg",1)#元画像読み込み
35#mask_and_alpha(img, maskgenerate(img))
36#切り取り用
37count = 1
38clear = [0,0,0]
39while(1):
40    with open("frag\frag.txt",'rb') as frag:
41        data = pickle.load(frag)
42        if data[0] == '1':
43            count = str(count)
44            img = cv2.imread("base\" + data[1] + '000' + count + data[2], 1)
45            mask_and_alpha(img, maskgenerate(img,data[1],count,data[2]),data[1],count,data[2])
46        count = int(count)
47        count += 1
48
49    with open("frag\frag.txt",'wb') as file:
50        pickle.dump(clear, file)
51        
52
53