Integration for Weights and Biases

ZPdesu / Barbershop

Barbershop: GAN-based Image Compositing using Segmentation Masks (SIGGRAPH Asia 2021)

https://zpdesu.github.io/Barbershop/

MIT License

1.32k stars 264 forks source link

Integration for Weights and Biases #34

Open soumik12345 opened 2 years ago

soumik12345 commented 2 years ago

This PR includes the following features:

Streamlining main.py to include the logic from align_face.py.
Updated the command line interface for main.py.
Formatting the codebase using black.
Include a jupyter notebook for inference.
Weights and Biases integration:
- Displaying prediction results using Tables.
- Support for tracking configs.
- Dataset and models hosted as public Artifacts.

Link to Weights and Biases Dashboard: https://wandb.ai/geekyrakshit/barbershop

soumik12345 commented 2 years ago

Hi @ZPdesu can you please take a look at this PR?

soumik12345 commented 2 years ago

Hi @ZPdesu we look forward to some feedback from your end regarding this PR. We are also happy to share that we have published a paper summary of Barbershop on wandb.ai/fully-connected. We are also attaching some screenshots of our dashboard. It would be great if we could merge this PR following some feedback from your end.

Screenshot 2022-04-29 at 7 41 59 PM Screenshot 2022-04-29 at 7 43 46 PM

tranphong1991 commented 2 years ago

Hi @soumik12345, thanks for the notebook. I am looking through your inference.ipynb notebook, look like the latent files part are missing.

soumik12345 commented 2 years ago

Hi @soumik12345, thanks for the notebook. I am looking through your inference.ipynb notebook, look like the latent files part are missing.

Hi @tranphong1991, we are looking into the issue. We will fix it as soon as possible. Thanks for the feedback.