idealo/image-quality-assessment
A deep learning system that predicts aesthetic and technical quality of images using transfer-learned CNNs.

Velocity · 7d
+0.8
★ / day
Trend
→steady
star history
This repository provides Keras/TensorFlow implementations of Google’s NIMA paper for neural image assessment. It uses pre-trained ImageNet models fine-tuned to classify image quality on AVA and TID2013 datasets. The system consists of two models—one for aesthetic quality and one for technical quality—which are trained via transfer learning. Docker support is provided for both CPU and GPU training on AWS EC2, and pre-trained models are included.