Integration for Weights and Biases

[soumik12345]

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]

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

[soumik12345]

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.

[tranphong1991]

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

[soumik12345]

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.