This folder contains demo notebooks for the TensorFlow UNet Industrial model.
### 1. TensorFlow_UNet_Industrial_TF_train_and_inference.ipynb: end to end training and inference demo.
The most convenient way to make use of the NVIDIA Tensorflow UNet model is via a docker container, which provides a self-contained, isolated and re-producible environment for all experiments. Refer to the [Quick Start Guide section](https://github.com/vinhngx/DeepLearningExamples/tree/vinhn_unet_industrial_demo/TensorFlow/Segmentation/UNet_Industrial#requirements) of the Readme documentation for a comprehensive guide. We briefly summarize the steps here.
where `/path/to/dataset` is the path on the host machine where the data was/is to be downloaded. More on data set preparation in the next section. `/path/to/results` is wher the trained model will be stored.
Within the docker interactive bash session, start Jupyter with
```
jupyter notebook --ip 0.0.0.0 --port 8888
```
Then open the Jupyter GUI interface on your host machine at http://localhost:8888. Within the container, this notebook itself is located at `/workspace/unet_industrial/notebooks`.
### 2. Colab_UNet_Industrial_TF_TFTRT_inference_demo.ipynb: inference from a pretrained UNet model with TensorFlow-TensorRT (TF-TRT).
This notebook is designed to run on Google Colab via this [link](https://colab.research.google.com/github/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/Segmentation/UNet_Industrial/notebooks/Colab_UNet_Industrial_TF_TFTRT_inference_demo.ipynb)
### 3. Colab_UNet_Industrial_TF_TFHub_export.ipynb: Colab notebook demostrating creation of TF-Hub module from NVIDIA NGC UNet model.
This notebook is designed to run on Google Colab vie this [link](https://colab.research.google.com/github/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/Segmentation/UNet_Industrial/notebooks/Colab_UNet_Industrial_TF_TFHub_export.ipynb)
This notebook is designed to run on Google Colab vie this [link](https://colab.research.google.com/github/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/Segmentation/UNet_Industrial/notebooks/Colab_UNet_Industrial_TF_TFHub_inference_demo.ipynb)