← all repositories

idealo/image-quality-assessment

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

2.2k stars Python Computer Vision
image-quality-assessment
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.