Skip to content

This project utilizes several machine learning algorithms to performs comparisons between a base image set used for training and a comparison image. AUC and MSE will be used as the primary metrics for measuring comparison.

License

Notifications You must be signed in to change notification settings

mattlitz/ml-image-comparison-tool

Repository files navigation

ml-image-comparison-tool

This project utilizes several machine learning algorithms to performs comparisons between a base image set used for training and a comparison image. AUC and MSE will be used as the primary metrics for measuring comparison.

The following Deep Learning methods are utilized for binary classification:

  • CNN
  • Transfer Learning with ResNet 50 and EfficientNet b0
  • ViT - Vision Transformer
  • Binary Classification with AutoEncoder

About

This project utilizes several machine learning algorithms to performs comparisons between a base image set used for training and a comparison image. AUC and MSE will be used as the primary metrics for measuring comparison.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages