Skip to content

Releases: titu1994/neural-image-assessment

Inception ResNet V2 Weights

10 Jan 04:21
Compare
Choose a tag to compare

Courtesy of @tfriedel, weights for Inception ResNet v2 which were fine-tuned on the AVA dataset.

Final score on val set ~ 0.07 (reported by @tfriedel)

NASNet Mobile Weights

07 Jan 06:10
Compare
Choose a tag to compare

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

06 Jan 04:52
Compare
Choose a tag to compare

Weights finetuned from the model from v0.2, using the proper loss from #3

MobileNet Weights

05 Jan 11:18
Compare
Choose a tag to compare

Weights trained on fix from #2

MobileNet Weights

02 Jan 15:16
Compare
Choose a tag to compare

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.