Skip to content

Improved development features for new models

Choose a tag to compare

@lovaslin lovaslin released this 16 Dec 08:01
· 23 commits to main since this release

This release include new/improved development feature to make training new models easier.

List of changes:

  • Improved function to generate dataset for training/testing new models (defect_detection.generate_dataset)
    - Supports taking images from different sources and combined them into one dataset
    - Improved data augmentation feature

  • Improved function to load a batch of images from disk directly to the required device (defect_detection.load_batch)
    - Support option to automatically generate a noisy image batch (useful for training new models)

  • Added function to initialize and train a new model on a given dataset (defect_detection.deepAE_train)
    - The structure of the model must be defined in a configuration file
    - The provided training loop supports automatic input noising, learning rate decay and training/validation split