unet的tensorflow代码_UNet-in-Tensorflow

U-Net Implementation in TensorFlow

Re implementation of U-Net in Tensorflow

to check how image segmentations can be used for detection problems

Original Paper

Summary

Vehicle Detection using U-Net

Objective: detect vehicles

Find a function f such that y = f(X)

Input

Shape

Explanation

Example

X: 3-D Tensor

(640, 960, 3)

RGB image in an array

y: 3-D Tensor

(640, 960, 1)

Binarized image. Bacground is 0

vehicle is masked as 255

Loss function: maximize IOU

(intersection of prediction & grount truth)

-------------------------------------------

(union of prediction & ground truth)

Examples on Test Data: tr