nima.pytorch icon indicating copy to clipboard operation
nima.pytorch copied to clipboard

Update to modern pytorch and fix model ckpt link

Open Dawars opened this issue 3 years ago • 1 comments

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)

Dawars avatar Jun 14 '21 16:06 Dawars

where can i download the pretrained-model

shineYuSong avatar Jun 29 '22 04:06 shineYuSong