Releases: titu1994/neural-image-assessment
Releases · titu1994/neural-image-assessment
Inception ResNet V2 Weights
NASNet Mobile Weights
Weights from NASNet Mobile model which have just been pre-trained on the AVA dataset.
Note : There are two files.
nasnet_weights.h5
is trained fully, and scores 0.067.nasnet_weights_0_0848.h5
is for the pre-trained version, with scores of 0.0848.
Notable limitations :
- Requires images to be 224x224 size only during evaluation, unlike MobileNet which can take any image size.
MobileNet Weights
Weights finetuned from the model from v0.2, using the proper loss from #3
MobileNet Weights
Weights trained on fix from #2
MobileNet Weights
Weights for MobileNet 224, trained for 15 epochs on the AVA dataset.
Note : Model was trained incorrectly using wrong loss. Use from tag v0.2 or tag v0.2.1 above for correct scores, which is trained on correct loss.