import cv2
if __name__ == "__main__":
if len(argv) < 2:
print("Provide an index")
test_img_index = int(argv[1])
img = cv2.imread(f"test_data/frame_{test_img_index}.png", cv2.IMREAD_COLOR)
game_data_handler = TestGameData(test_img_index)
game_classifier = GameClassifier()
game_classifier.load()
digit_classifier = DigitClassifier()
digit_classifier.load()
game_state = GameState(game_data_handler, digit_classifier)
state, data = game_state.get_game_state(img)
game_data, my_team = data
classifier_input = game_dataset.shape_input(game_data, game_data_handler)
outcome = game_classifier.predict(classifier_input)
if my_team == "red":
outcome = 1 - outcome
pct = f"{outcome * 100:.2f}"
print(f"Probability of win: {pct}%", flush=True)
shape = (img.shape[1] // 2, img.shape[0] // 2)
resized = cv2.resize(img, shape, interpolation=cv2.INTER_AREA)
{
"championId": self.champions[8], "teamId": 200, "summonerId": self.summ_ids[8],
"spell1Id": self.get_summoner_spell_index(self.summs[8][0]), # Exhaust
"spell2Id": self.get_summoner_spell_index(self.summs[8][1]) # Flash
},
{
"championId": self.champions[9], "teamId": 200, "summonerId": self.summ_ids[9],
"spell1Id": self.get_summoner_spell_index(self.summs[9][0]), # Ignite
"spell2Id": self.get_summoner_spell_index(self.summs[9][1]) # Flash
}
]
}
if __name__ == "__main__":
test_img_index = 4
img = cv2.imread(f"test_data/frame_{test_img_index}.png", cv2.IMREAD_COLOR)
champion_data = TestGameData(test_img_index)
digit_classifier = DigitClassifier()
digit_classifier.load()
game_state_handler = GameState(champion_data, digit_classifier)
state, data = game_state_handler.get_game_state(img)
game_data, my_team = data
for team in game_data:
print(f"====== {team.upper()} TEAM ======")
print(f"Towers destroyed: {game_data[team]['towers_destroyed']}")
print(f"Dragons: {game_data[team]['dragons']}")
for player_data in game_data[team]["players"]:
print(player_data)
print("***********************************************")
