This basic purpose of this project is to make an app that will make a segmentation of an input image. To accomplish this U-Net model was trained on a Segmentation Full Body TikTok Dancing Dataset from Kaggle.
Project constists of Colab Notebook and a Website that are located in different branches.
Firsly, make sure that you have installed all requirements.
Also preferably to have following versions of CUDA 11.1 and Python 3.7.13.
Than you can clone this repo and you will have to change several pathes in code, after this you are done and you can run notebook or website.