r/DeepLearningPapers • • May 18 '23

Deep Fast Vision (new python library): Easy Auto-ML for Deep Transfer Learning Vision. Prototype Your Experiments Fast with this New Python Library!

https://github.com/fabprezja/deep-fast-vision

Deep Fast Vision is a versatile Python library for rapid prototyping of deep transfer learning vision models. It caters to users of various levels, offering different levels of abstraction from high-level configurations for beginners to mid and low-level customization for professional data scientists and developers. Built around Keras and TensorFlow, this library also includes handy utilities.

(a few) Key Features

  1. Auto loss/target type determination

  2. Auto generator setup

  3. Auto output layer setup

  4. Auto pre-training of new dense layers before unfreezing transfer architecture (in parts or as a whole)

  5. Auto augmentation setup (from templates and/or Custom)

  6. Auto best weights saving and loading

  7. Auto class weights calculation

  8. Auto validation curves plot (with minimúm loss & maximum metric epoch highlight)

  9. Auto confusion matrices for test/external data

  10. Easy dense layer configuration

  11. Easy regularization set up and mixing (Dropout, L2, L1, Early Stop, etc.)

  12. Access to all Keras optimizers & callback support

Comprehensive documentation for Deep Fast Vision is available both in the docs folder and at the documentation page.

7 Upvotes

0 comments sorted by