search

google-research / mint

Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.
Apache License 2.0
497 stars 85 forks source link

How can I visualize results of evaluation? #57

Open euneestella opened 1 year ago

euneestella commented 1 year ago

Hi, thanks for your amazing work!

I successfully ran evaluator.py and calculated FIDs with tools/calculate_scores.py. I want to visualize the evaluation results by using function named visualize which is defined as below. https://github.com/google-research/mint/blob/b8f8bdfbbe3fbfa67831a2ef9bcf71a6a9e74552/tools/calculate_scores.py#L211 https://github.com/google-research/mint/blob/b8f8bdfbbe3fbfa67831a2ef9bcf71a6a9e74552/tools/calculate_scores.py#L54-L79

However, my code says Aborted (core dumped) when I uncomment L211 of tools/calculate_scores.py and doesn't work. Can you help me out with this problem?