INNER CODE UNIT · Python

image

zuruoke/watermark-removal · guided_batch_test.py:80

        image = image[:h//grid*grid, :w//grid*grid, :]
        mask = mask[:h//grid*grid, :w//grid*grid, :]
        guidance = guidance[:h//grid*grid, :w//grid*grid, :]
        print('Shape of image: {}'.format(image.shape))

        image = np.expand_dims(image, 0)
        guidance = np.expand_dims(guidance, 0)
        mask = np.expand_dims(mask, 0)
        input_image = np.concatenate([image, guidance, mask], axis=2)

        # load pretrained model
        result = sess.run(output, feed_dict={input_image_ph: input_image})
        print('Processed: {}'.format(out))
        cv2.imwrite(out, result[0][:, :, ::-1])

    print('Time total: {}'.format(time.time() - t))

View source record →