search

ternaus / cloths_segmentation

Code for binary segmentation of cloths
MIT License
222 stars 40 forks source link

post processing #6

Open sudhakar-sah opened 3 years ago

sudhakar-sah commented 3 years ago

thank you for this amazing repo. the model works really well. I have one query, are you using some sort of heuristics/post-processing for the demo. The model output and the web demo output do not match in many images (web app work very well actually compared to the pre-trained model).

sree3333 commented 3 years ago

Hey @sudhakar-sah, I am having NCCL issues running the model. Could you please let me know how did you set up the model to run?

sudhakar-sah commented 3 years ago

@sree3333 NCCL error is related to distributed training/inference. You can just follow the google colab code for inference. three wont be any NCCL error.

sree3333 commented 3 years ago

Thanks for writing me back @sudhakar-sah. My issue got resolved with NCCL but I now have issues with training. Have you tried training on this model, because I am getting errors running the training. Please let me know if you have any success starting the training process