truskovskiyk / nima.pytorch

NIMA: Neural IMage Assessment
MIT License
335 stars 79 forks source link

Update to modern pytorch and fix model ckpt link #30

Open Dawars opened 3 years ago

Dawars commented 3 years ago

I fixed up the code to make inference possible out of the box

Inference example:

from nima.inference.inference_model import InferenceModel

model = InferenceModel(path_to_model=Path('./pretrain-model.pth'))

result = model.predict_from_pil_image(img)
shineYuSong commented 2 years ago

where can i download the pretrained-model