Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(metric): 🌟 add NIMA #49

Merged
merged 13 commits into from
Aug 26, 2022
Merged

feat(metric): 🌟 add NIMA #49

merged 13 commits into from
Aug 26, 2022

Conversation

kargaranamir
Copy link
Collaborator

First I tried this code, but I couldn't find any pre-trained models for it. According to the code it uses one of the ResNet models. However, most of the codes on the Internet use the previous version of this repo, and the maintainers of this project assure that the previous version is still valid and working. The code that I use is based on MobileNetv2, and the pre-trained model URL can be found here.

It seems the deployed model on the Heroku app is different from the pre-trained one (issue8, issue28)

@kargaranamir kargaranamir requested a review from mplaine June 5, 2022 14:44
@kargaranamir
Copy link
Collaborator Author

kargaranamir commented Jun 16, 2022

DENSENET architecture gets a better result according to this repo. Also, we tested NIMA's different architectures on the website aesthetic dataset provided by labinthewild and came to the same conclusion. The current model downloaded from here.

@mplaine
Copy link
Collaborator

mplaine commented Aug 26, 2022

Minor tasks:

  • magic numbers -> constants in m18_nima.py
  • assign m18 to a category in metrics.json
  • add a score section for m18_1
  • run evaluator and update histogram images and score details for m18 in metrics.json

@mplaine mplaine merged commit 936c1f4 into aim2 Aug 26, 2022
@mplaine mplaine deleted the feat/NIMA branch August 26, 2022 18:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants