(reference_poster_folder, generated_poster_img_folder, clip=False)
| 448 | return fid_score |
| 449 | |
| 450 | def compute_fid(reference_poster_folder, generated_poster_img_folder, clip=False): |
| 451 | if clip: |
| 452 | return compare_folders_with_clip(reference_poster_folder, generated_poster_img_folder) |
| 453 | model = fid.InceptionV3([fid.InceptionV3.BLOCK_INDEX_BY_DIM[2048]]).to('cuda') |
| 454 | |
| 455 | m1, s1 = compute_statistics_of_path(reference_poster_folder, model, 1, 2048, 'cuda') |
| 456 | m2, s2 = compute_statistics_of_path(generated_poster_img_folder, model, 1, 2048, 'cuda') |
| 457 | |
| 458 | fid_score = fid.calculate_frechet_distance( |
| 459 | m1, s1, m2, s2 |
| 460 | ) |
| 461 | |
| 462 | return fid_score |
| 463 | |
| 464 | |
| 465 | def get_poster_text(poster_path, check_fail=True): |
nothing calls this directly
no test coverage detected