tensorflow agrega ruido a las imágenes de entrenamiento

importar cv2

importar os

importar numpy como np

os.environ["CUDA_VISIBLE_DEVICES"] = " 3"

path="/home/ta/Desktop/gs_for_test/1.bmp"

ori=cv2.imread(ruta, cv2.IMREAD_GRAYSCALE)

img=tf.convert_to_tensor(ori)

ruido=tf.random_normal(shape=img.shape, mean=0, stddev=50)

drop_noise=tf.nn.dropout(ruido, 0.01 )

final=tf.add(img, tf.cast(drop_noise, dtype=tf.uint8))

cv2.namedWindow('img', cv2.WINDOW_KEEPRATIO) p>

cv2.namedWindow('final', cv2.WINDOW_KEEPRATIO)

cv2.KEEPRATIO).

Utiliza tf.Session() como sesión:

fimg=sess.run(final)

img__=sess.run(img)

cv2.imshow("final", fimg)

cv2.imshow("img", img__)

cv2.waitKey()