-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathChangeDataBackground.py
153 lines (122 loc) · 6.91 KB
/
ChangeDataBackground.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
import json
import matplotlib.pyplot as plt
import cv2
import os
from labelme import utils
import numpy as np
from base64 import b64encode
from json import dumps
import time
import random
from skimage.util import random_noise
import base64
import json
import re
from copy import deepcopy
import argparse
import math
class DataAugmentation():
def __init__(self):
self.rename = 'obj_data'
self.obj_number = 1000
self.cnt = 0
self.DIR = '/home/chien/Data_Augmentation/background'
self.PNG = '/home/chien/Data_Augmentation/png'
self.Data = '/home/chien/Data_Augmentation/Extend_data'
self.datanames1 = os.listdir('/home/chien/Data_Augmentation/Extend_data')
self.datanames2 = os.listdir('/home/chien/Data_Augmentation/background')
self.json_data = np.zeros([self.obj_number,1]).astype(np.str)
self.background_data = np.zeros([self.obj_number,1]).astype(np.str)
self.oldbackground_data = np.zeros([self.obj_number,1]).astype(np.str)
self.output_data = np.zeros([self.obj_number,1]).astype(np.str)
self.json_name=np.zeros([self.obj_number,1]).astype(np.str)
i = 0
j = 0
m = 0
for dataname in self.datanames1:
if os.path.splitext(dataname)[1] == '.json':#目录下包含.json的文件
self.json_data[i]=dataname
self.json_name[i], suffix = os.path.splitext(",".join(self.json_data[i]))
i+=1
for dataname in self.datanames2:
if os.path.splitext(dataname)[1] == '.jpg':#目录下包含.jpg的文件
self.background_data[j]=dataname
j+=1
for dataname in self.datanames1:
if os.path.splitext(dataname)[1] == '.jpg':#目录下包含.jpg的文件
self.oldbackground_data[m]=dataname
m+=1
def _changeLight(self, img):
alpha = random.uniform(0.35, 1)
blank = np.zeros(img.shape, img.dtype)
return cv2.addWeighted(img, alpha, blank, 1 - alpha, 0)
def generate_data(self):
for k in range(len([name for name in os.listdir(self.DIR) if os.path.isfile(os.path.join(self.DIR, name))])):
for j in range(int(len(([name for name in os.listdir(self.Data) if os.path.isfile(os.path.join(self.Data, name))]))/2)):
json_file= self.Data +'/{}.json'.format(",".join(self.json_name[j]))
data = json.load(open(json_file))
img = utils.img_b64_to_arr(data['imageData'])
label_name_to_value = {'_background_': 0}
for shape in data['shapes']:
label_name = shape['label']
if label_name in label_name_to_value:
label_value = label_name_to_value[label_name]
else:
label_value = len(label_name_to_value)
label_name_to_value[label_name] = label_value
lbl, lbl_names = utils.shapes_to_label(img.shape, data['shapes'], label_name_to_value)
mask=[]
class_id=[]
for i in range(1,len(label_name_to_value)):
mask.append((lbl==i).astype(np.uint8))
class_id.append(i)
mask=np.asarray(mask,np.uint8)
mask=np.transpose(np.asarray(mask,np.uint8),[1,2,0])
all_mask=0
for i in range(0,len(class_id)):
retval, im_at_fixed = cv2.threshold(mask[:,:,i], 0, 255, cv2.THRESH_BINARY)
all_mask = all_mask + im_at_fixed
cv2.imwrite( self.PNG + "/{}.png".format(",".join(self.json_name[j])), all_mask)
label_together = cv2.imread( self.PNG + '/{}.png'.format(",".join(self.json_name[j])))
oldbackground = cv2.imread( self.Data +'/{}.jpg'.format(",".join(self.json_name[j]))) #oldbackground_data-->json_name
oldbackground = cv2.resize(oldbackground,(640,480))
oldbackground = cv2.cvtColor(oldbackground,cv2.COLOR_BGR2RGB)
Together = cv2.bitwise_and(label_together,oldbackground) #两个图片一个是彩色，另一个是彩色，不能做位与运算，需要将第一个图灰度话后再操作
newbackground = cv2.imread( self.DIR + '/{}'.format(",".join(self.background_data[k]))) # 读取图片img2
newbackground = cv2.cvtColor(newbackground,cv2.COLOR_BGR2RGB)
newbackground = self._changeLight(newbackground)
[x,y,z]=np.shape(Together)
newbackground = cv2.resize(newbackground,(640,480)) # 为图片重新指定尺寸
imgTogether = Together + newbackground
Inverse_label_together = 255 - label_together
Remove_label_together = cv2.bitwise_and(Inverse_label_together,newbackground)
imgnew1Together = Together + Remove_label_together
imgnew1Together = cv2.cvtColor(imgnew1Together,cv2.COLOR_BGR2RGB)
plt.imshow(imgnew1Together)
#cv2.imwrite( 'output_data/' + "/{}.jpg".format(",".join(self.json_name[j])),imgnew1Together)
cv2.imwrite( 'output_data/' + "/{}_{}.jpg".format(self.rename, self.cnt + 1),imgnew1Together)
ENCODING = 'utf-8'
#IMAGE_NAME = '{}.jpg'.format(",".join(self.json_name[j]))
#JSON_NAME = '{}.json'.format(",".join(self.json_name[j]))
IMAGE_NAME = '{}_{}.jpg'.format(self.rename, self.cnt + 1)
JSON_NAME = '{}_{}.json'.format(self.rename, self.cnt + 1)
with open('output_data/'+ IMAGE_NAME, 'rb') as jpg_file:
byte_content = jpg_file.read()
base64_bytes = b64encode(byte_content)
base64_string = base64_bytes.decode(ENCODING)
raw_data = {}
raw_data["version"] = "4.5.6"
raw_data["shapes"] = data['shapes']
raw_data["imagePath"] = IMAGE_NAME
raw_data["imageData"] = base64_string
raw_data["imageHeight"] = 480
raw_data["imageWidth"] = 640
jsondata = dumps(raw_data, indent=2)
with open('output_data/'+ JSON_NAME, 'w') as json_file:
json_file.write(jsondata)
print('Generating dataset : {}_{}.jpg'.format(self.rename, self.cnt + 1))
print('Generating dataset : {}_{}.json'.format(self.rename, self.cnt + 1) )
self.cnt+=1
if __name__ == '__main__':
data = DataAugmentation()
data.generate_data()