Skip to content

This project uses annotated images of soles from the shoe to train the model with UNET architecture in order to obtain a mask with predicted sole outline and location

License

Notifications You must be signed in to change notification settings

marseller/Shoe-sole-segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Shoe-Sole Segmentation UNET (PyTorch)

This project uses annotated images of soles from the shoe to train the model with UNET architecture in order to obtain a mask with predicted sole outline and location.This approach can be used in the shoe recycling industry to locate the sole and separate it from the upper part of the shoe, which would be an alternative method of sepparating the rubber part of the show within shoe-recycling industry.
Implemented in PyTorch.
Used dataset : UT Zappos50K - https://vision.cs.utexas.edu/projects/finegrained/utzap50k/

model_example_output output

About

This project uses annotated images of soles from the shoe to train the model with UNET architecture in order to obtain a mask with predicted sole outline and location

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published