triangleczh / cs175deeplearning Goto Github PK

When I run batch_x_np, batch_joints_np = sess.run(next_ele) in run_train.py，a mistake has occurred:tensorflow.python.framework.errors_impl.InvalidArgumentError: Input to reshape is a tensor with 0 values, but the requested shape has 10

Github not include to this (from noiseFilter import gasuss_noise,sp_noise,horizontal_flip))

so i find

def sp_noise(image,prob):
#Add salt and pepper noise to image, prob: Probability of the noise
output = np.zeros(image.shape,np.uint8)
thres = 1 - prob
for i in range(image.shape[0]):
for j in range(image.shape[1]):
rdn = random.random()
if rdn < prob:
output[i][j] = 0
elif rdn > thres:
output[i][j] = 255
else:
output[i][j] = image[i][j]
return output

def add_gaussian_noise(X_imgs):
gaussian_noise_imgs = []
row, col, _ = X_imgs[0].shape
# Gaussian distribution parameters
mean = 0
var = 0.1
sigma = var ** 0.5

for X_img in X_imgs:
	gaussian = np.random.random((row, col, 1)).astype(np.float32)
	gaussian = np.concatenate((gaussian, gaussian, gaussian), axis = 2)
	gaussian_img = cv2.addWeighted(X_img, 0.75, 0.25 * gaussian, 0.25, 0)
	gaussian_noise_imgs.append(gaussian_img)
gaussian_noise_imgs = np.array(gaussian_noise_imgs, dtype = np.float32)
return gaussian_noise_imgs