m.giller/ray-the-ripper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added missing reference normalization

Browse source
This commit is contained in:
Maximilian Giller 2025-09-06 19:15:43 +02:00
parent b02c1b3228
commit bb5092f700
1 changed files with 28 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

56
src/label_episodes.py
View file

@ -230,53 +230,53 @@ if __name__ == "__main__":
# "./episodes/B1_t09.mkv", # "./episodes/B1_t09.mkv",
# "./episodes/B1_t10.mkv", # "./episodes/B1_t10.mkv",
] ]
references = { references = {
"1" : [ "1": [
cv2.imread("./1/9hYDr40VIWbAHGjmlpSkyqlDaVQ.webp"), normalize_image(cv2.imread("./1/9hYDr40VIWbAHGjmlpSkyqlDaVQ.webp")),
cv2.imread("./1/cmYUHroQ9zRhOYiTf0ozMWe1Bzs.webp"), normalize_image(cv2.imread("./1/cmYUHroQ9zRhOYiTf0ozMWe1Bzs.webp")),
cv2.imread("./1/yLwaP7Y7O3Wyrht9lPmTDk7LSuG.webp"), normalize_image(cv2.imread("./1/yLwaP7Y7O3Wyrht9lPmTDk7LSuG.webp")),
], ],
"2" : [ "2": [
cv2.imread("./2/g2pyyFRFBS18jYByRdiunWr7nap.webp"), normalize_image(cv2.imread("./2/g2pyyFRFBS18jYByRdiunWr7nap.webp")),
cv2.imread("./2/wKCm3I7efXfsWGNrG28d3cIsL9B.webp"), normalize_image(cv2.imread("./2/wKCm3I7efXfsWGNrG28d3cIsL9B.webp")),
cv2.imread("./2/yhL1JBnTDK30BtPbr8gZ5ZXCASL.webp"), normalize_image(cv2.imread("./2/yhL1JBnTDK30BtPbr8gZ5ZXCASL.webp")),
], ],
"3" : [ "3": [
cv2.imread("./3/d9F2x01XNhi65GTXwQjrdDJKgqm.webp"), normalize_image(cv2.imread("./3/d9F2x01XNhi65GTXwQjrdDJKgqm.webp")),
cv2.imread("./3/kV7VAvLgjjhPTI56OnNvcNaHjDM.webp"), normalize_image(cv2.imread("./3/kV7VAvLgjjhPTI56OnNvcNaHjDM.webp")),
cv2.imread("./3/mLxkumagHSYg4KNLcEIgHpiJrx.webp"), normalize_image(cv2.imread("./3/mLxkumagHSYg4KNLcEIgHpiJrx.webp")),
], ],
"4" : [ "4": [
cv2.imread("./4/9q62zw89SJiRD3kjNuhUIswWHTa.webp"), normalize_image(cv2.imread("./4/9q62zw89SJiRD3kjNuhUIswWHTa.webp")),
cv2.imread("./4/lyi4XUUssCacaEQs2FCGSDCKDUK.webp"), normalize_image(cv2.imread("./4/lyi4XUUssCacaEQs2FCGSDCKDUK.webp")),
], ],
"5" : [ "5": [
cv2.imread("./5/uIztge10KNlRU4KUFNXWPGCiG8e.webp"), normalize_image(cv2.imread("./5/uIztge10KNlRU4KUFNXWPGCiG8e.webp")),
cv2.imread("./5/vuhdflodFfPkvekVWbS4JZMYHa7.webp"), normalize_image(cv2.imread("./5/vuhdflodFfPkvekVWbS4JZMYHa7.webp")),
cv2.imread("./5/xmvi4bBk5TQyuL6tpiYvNoIPft.webp"), normalize_image(cv2.imread("./5/xmvi4bBk5TQyuL6tpiYvNoIPft.webp")),
], ],
"6" : [ "6": [
cv2.imread("./6/2I7WgvRwnYDNjbiztvbOAldYGde.webp"), normalize_image(cv2.imread("./6/2I7WgvRwnYDNjbiztvbOAldYGde.webp")),
cv2.imread("./6/dIaBagmUbP1AO6xFQzfQud5lk4I.webp"), normalize_image(cv2.imread("./6/dIaBagmUbP1AO6xFQzfQud5lk4I.webp")),
cv2.imread("./6/ekDZrFRDJDyx78uttMyVT9mPhjL.webp"), normalize_image(cv2.imread("./6/ekDZrFRDJDyx78uttMyVT9mPhjL.webp")),
], ],
} }
results = match_episodes_to_references(episodes, references) results = match_episodes_to_references(episodes, references)
print(results) print(results)
print("\nBy Episode:") print("\nBy Episode:")
for ep, scores in results.items(): for ep, scores in results.items():
print(f"{ep}") print(f"{ep}")
for ref, value in scores.items(): for ref, value in scores.items():
print(f"\t{ref}\t{value}") print(f"\t{ref}\t{value}")
print("\nBy Reference:") print("\nBy Reference:")
for ref in list(results.values())[0].keys(): for ref in list(results.values())[0].keys():
print(f"{ref}") print(f"{ref}")
for episode, scores in results.items(): for episode, scores in results.items():
print(f"\t{episode}\t{scores[ref]}") print(f"\t{episode}\t{scores[ref]}")
print("Done.") print("Done.")