Images

PyTorch NIMA: Neural IMage Assessment

PyTorch NIMA: Neural IMage Assessment

PyTorch implementation of Neural IMage Assessment by Hossein Talebi and Peyman Milanfar. You can learn more from this post at Google Research Blog.

Installing

Docker

docker run -it truskovskiyk/nima:latest /bin/bash

PYPI package (In Progress)

pip install nima

VirtualEnv

git clone https://github.com/truskovskiyk/nima.pytorch.git
cd nima.pytorch
virtualenv -p python3.7 env
source ./env/bin/activate

Dataset

The model was trained on the AVA (Aesthetic Visual Analysis) dataset
You can get it from here
Here are some examples of images with theire scores
result1

Pre-train model (In Progress)

Deployment (In progress)

Usage

nima-cli


Usage: cli.py [OPTIONS] COMMAND [ARGS]...


Options:

  --help  Show this message and exit.


Commands:

  get_image_score  Get


 


 




 


To finish reading, please visit source site