https://github.com/pinto0309/segnet-tf
Tensorflow implementation of SegNet Tensorflow 1.11.0 + Python (I made minor bugfixes for toimcio/SegNet-tensorflow)
https://github.com/pinto0309/segnet-tf
python segnet semantic-segmentation tensorflow
Last synced: about 2 months ago
JSON representation
Tensorflow implementation of SegNet Tensorflow 1.11.0 + Python (I made minor bugfixes for toimcio/SegNet-tensorflow)
- Host: GitHub
- URL: https://github.com/pinto0309/segnet-tf
- Owner: PINTO0309
- Created: 2018-10-08T15:54:58.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2018-10-09T15:37:23.000Z (over 7 years ago)
- Last Synced: 2024-10-29T08:05:21.807Z (over 1 year ago)
- Topics: python, segnet, semantic-segmentation, tensorflow
- Language: Jupyter Notebook
- Homepage:
- Size: 2.63 MB
- Stars: 1
- Watchers: 3
- Forks: 2
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# SegNet-Tensorflow implementation
[October 09, 2018]
I made minor bugfixes for toimcio/SegNet-tensorflow.
https://github.com/toimcio/SegNet-tensorflow
# Usage
## 1. Training
```
$ cd ~
$ git clone https://github.com/PINTO0309/SegNet-TF.git
$ cd SegNet-TF
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=12wakrs1SSLTL50LuSibMTBpho_JySqFk" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=12wakrs1SSLTL50LuSibMTBpho_JySqFk" -o vgg16.npy
$ cd SegNet
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1FgMelph4IQOrjs3b3TfuZE9uczo-Ex_6" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1FgMelph4IQOrjs3b3TfuZE9uczo-Ex_6" -o CamVid.tar.gz
$ tar -zxvf CamVid.tar.gz
$ rm CamVid.tar.gz
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1j16hiO2-9BRXKaVTGYKmrwQnN2yIkgw1" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1j16hiO2-9BRXKaVTGYKmrwQnN2yIkgw1" -o sun3d_dataset.tar.gz
$ tar -zxvf sun3d_dataset.tar.gz
$ rm sun3d_dataset.tar.gz
$ sudo -H pip3 install tensorflow-gpu==1.11
$ cd ..
$ python3
>> from SegNet import SegNet
>> SegNet().train()
```
## 2. Slimming weight files for deployment
A weight file after slimming is generated under the "ckpt" folder.
```
$ python3 SegNetInfer.py
```
## 3. Freeze graph
```
$ python3 freeze_graph.py \
--input_graph=ckpt/deployfinal.pbtxt \
--input_checkpoint=ckpt/deployfinal.ckpt \
--output_graph=ckpt/deployfinal.pb \
--output_node_names=conv_classifier/output \
--input_binary=False
```
## 4. Inference Test
```
$ python3
>> from SegNet import SegNet
>> SegNet().visual_results()
```
# Environment
Requirement: Tensorflow-GPU 1.11.0
Ubuntu: 16.04
python: 3.5.2
CUDA: 9.0
cuDNN: 7
GPU:Geforce GTX 1070
# Material
## 1. Check Point File (.ckpt)
```
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1vFa6h4SkdJ6irwUwnbSJUFf3tkH4Ina5" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1vFa6h4SkdJ6irwUwnbSJUFf3tkH4Ina5" -o deploy.tar.gz
```
## 2. Slimmed Check Point File (.ckpt)
```
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1WY98-AXRbo83r3z_5LktLtdiRV0w9IA1" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1WY98-AXRbo83r3z_5LktLtdiRV0w9IA1" -o deployfinal.tar.gz
```
## 3. Protocol Buffer Text (.pbtxt)
```
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1-eaByvjJAUvIdaS2Y3--KatCY0O3v24q" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1-eaByvjJAUvIdaS2Y3--KatCY0O3v24q" -o deployfinal.pbtxt
```
## 4. Frozen graph (.pb)
```
$ curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1N1xx7wo7qmmM3CVnqmYQU5dp-aFWGh6t" > /dev/null
$ CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
$ curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1N1xx7wo7qmmM3CVnqmYQU5dp-aFWGh6t" -o deployfinal.pb
```