Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Image generation support #135

Merged
merged 43 commits into from
Feb 27, 2023
Merged

Image generation support #135

merged 43 commits into from
Feb 27, 2023

Conversation

bcebere
Copy link
Contributor

@bcebere bcebere commented Feb 21, 2023

Description

  • Add image dataloader, with the ability to resize the images internally to predefined shapes.
  • Extend the core plugin for non-tabular inputs
  • Add ConvNet classifier support, with some predefined architectures.
  • Add Generator and Discriminators with predefined architectures for various image input sizes.
  • Add Image GAN support on top of the Generators and Discriminators.
  • Add Conditional GAN support for the Generators and Discriminators
  • New metrics: FID score
  • Adapt the detection and performance metrics for images
  • Add Image Conditional GAN plugin
  • Add Image AdsGAN plugin
  • Add tutorial on top of MedMNIST
  • Update docs and README

fixes #132

Checklist

@bcebere bcebere added the enhancement New feature or request label Feb 23, 2023
@bcebere bcebere changed the title [WIP] Image generation support Image generation support Feb 23, 2023
@bcebere bcebere merged commit 9a2ac98 into main Feb 27, 2023
@bcebere bcebere deleted the image_support branch February 27, 2023 06:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Image generation initial pipeline
2 participants