bdzyubak / tensorflow-sandbox

A repository for studying applications of Deep Learning across fields, and demonstrating samples of my code and project managment
0 stars 0 forks source link

Compare builtin tensorflow classification models #17

Closed bdzyubak closed 2 years ago

bdzyubak commented 2 years ago

CLASSIFICATION MODEL COMPARISON

Now that the module initializer and data fetcher are implemented, it is straightforward to test a variety of architectures against each other. Tensorflow Keras contains the following models ready for import. https://www.tensorflow.org/api_docs/python/tf/keras/applications

This enhancement request is to extend the module initializer to support all builtin classification models relevant to the image analysis tasks, create scripts for comparison, and pick the best model.

bdzyubak commented 2 years ago

All builtin functions listed on the above page are now supported. The script /home/m047659/Projects/Bogdans-Deep-Learning-Sandbox/KaggleHistopathologyDetection/train_compare_builtin_class_models.py will train all of them, with re-initialization to bypass local minima, for evaluation.