Skip to content

Object detection trained on deepfashion using icevision

License

Notifications You must be signed in to change notification settings

potipot/deepfashion

Repository files navigation

Binder

gradio-deployment

Gradio deployment

https://gradio.app/g/potipot/deepfashion

howto

to train the model you'll have to download the ds from https://github.com/switchablenorms/DeepFashion2 and place it in the datasets/train and datasets/validation directory.

The label failes are already cached and should be loaded automatically - check detector.ipynb to make sure.

Check the startup.sh to see what kind of preprocessing is necessary.

  1. Install nvidia apex:
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

Make sure your /usr/local/cuda points to cuda-10.2

  1. Create the env
conda env create -n deepfashion -f environment.yml
  1. Train
python detector.py

About

Object detection trained on deepfashion using icevision

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages