Returns 'image' with an extra channel set to all 1s.
(image)
| 409 | |
| 410 | |
| 411 | def wrap(image): |
| 412 | """Returns 'image' with an extra channel set to all 1s.""" |
| 413 | shape = tf.shape(image) |
| 414 | extended_channel = tf.ones([shape[0], shape[1], 1], image.dtype) |
| 415 | extended = tf.concat([image, extended_channel], 2) |
| 416 | return extended |
| 417 | |
| 418 | |
| 419 | def unwrap(image, replace): |
no outgoing calls
no test coverage detected