Light

nikhilroxtomar / unet-for-person-segmentation Goto Github PK

View Code? Open in Web Editor NEW

25.0 1.0 11.0 22.31 MB

U-Net for person segmentation in TensorFlow using Keras API.

Python 100.00%

tensorflow keras unet-image-segmentation unet-keras unet-segmentation

unet-for-person-segmentation's Introduction

Unet for Person Segmentation

We are using the famous UNet architecture for segmenting person from an image. For the person segmentation, we are going to use the person segmentation dataset. U-Net is built for Biomedical Image Segmentation. It is the base model for any segmentation task. It follows an encoder-decoder approach. It used skip connection to get the local information during downsampling path and use it during the upsampling path.

YouTube Video: https://youtu.be/qrL22HEaUGA
Arxiv Paper: U-Net: Convolutional Networks for Biomedical Image Segmentation

These images are generated after the model is trained on 2 epochs.

unet-for-person-segmentation's People

Contributors

Stargazers

Watchers

Forkers

ersaree amirunpri2018 eridanletalis sumesh1 dr-alok-tiwari sundawei junjie2008v apcc-geoslegend mcmarinica 19xx47 woojinjoung

unet-for-person-segmentation's Issues

Any chance you can share the pretrained model?

thanks!

Recall always jumping to zero from first epoch

Hi Nikhil,
Thanks for this UNET implementation.
I have followed exact steps in your tutorial and setup and gave all paths correctly.
I have used "train.py" and it is showing empty training data . are there any files me

Terminal log:
python train.py
Train: 4542 - 4542
Test: 1136 - 1136
Train for 379 steps, validate for 95 steps
Epoch 1/10
378/379 [============================>.] - ETA: 0s - loss: 0.0251 - mean_io_u: 1.0000 - recall: 0.0071 - precision: 0.2735
Epoch 00001: saving model to unet.h5
379/379 [==============================] - 273s 719ms/step - loss: 0.0251 - mean_io_u: 1.0000 - recall: 0.0071 - precision: 0.2735 - val_loss: 0.0086 - val_mean_io_u: 1.0000 - val_recall: 0.0000e+00 - val_precision: 0.0000e+00
Epoch 2/10
WARNING:tensorflow:Your input ran out of data; interrupting training. Make sure that your dataset or generator can generate at least steps_per_epoch * epochs batches (in this case, 3790 batches). You may need to use the repeat() function when building your dataset.

Epoch 00002: saving model to unet.h5
WARNING:tensorflow:Reduce LR on plateau conditioned on metric val_loss which is not available. Available metrics are: lr
WARNING:tensorflow:Early stopping conditioned on metric val_loss which is not available. Available metrics are:
0/379 [..............................] - ETA: 0sTraceback (most recent call last):
File "train.py", line 62, in
callbacks=callbacks
File "/home/anaconda3/envs/UNET/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/training.py", line 819, in fit
use_multiprocessing=use_multiprocessing)
File "/home/anaconda3/envs/UNET/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/training_v2.py", line 342, in fit
total_epochs=epochs)
File "/home/anaconda3/envs/UNET/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/training_v2.py", line 187, in run_one_epoch
aggregator.finalize()
File "/home/anaconda3/envs/UNET/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/training_utils.py", line 144, in finalize
raise ValueError('Empty training data.')
ValueError: Empty training data.

Recommend Projects

React

A declarative, efficient, and flexible JavaScript library for building user interfaces.
Vue.js

🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
Typescript

TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
TensorFlow

An Open Source Machine Learning Framework for Everyone
Django

The Web framework for perfectionists with deadlines.
Laravel

A PHP framework for web artisans
D3

Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

javascript

JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
web

Some thing interesting about web. New door for the world.
server

A server is a program made to process requests and deliver data to clients.
Machine learning

Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Visualization

Some thing interesting about visualization, use data art
Game

Some thing interesting about game, make everyone happy.

Recommend Org

Facebook

We are working to build community through open source technology. NB: members must have two-factor auth.
Microsoft

Open source projects and samples from Microsoft.
Google

Google ❤️ Open Source for everyone.
Alibaba

Alibaba Open Source for everyone
D3

Data-Driven Documents codes.
Tencent

China tencent open source team.