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("***********************************************")