def recoginize_deadly_weapons(self, payload):
h = payload['sample_height']
images = self._unpack_deadly_weapons_image(payload)
lang = payload['game_language']
deadly_weapon_recoginizer = DeadlyWeaponRecoginizer(lang)
votes = {}
for image in images:
weapon_id = deadly_weapon_recoginizer.predict(image)
votes[weapon_id] = votes.get(weapon_id, 0) + 1
best_result = (None, 0)
for weapon_id in votes.keys():
if votes[weapon_id] > best_result[1]:
best_result = (weapon_id, votes[weapon_id])
response_payload = {
'status': 'ok',
'deadly_weapon': best_result[0],
'score': best_result[1],
'total': len(images),
}
return response_payload
def recoginize_deadly_weapons(self, payload):
h = payload['sample_height']
images = self._unpack_deadly_weapons_image(payload)
lang = payload['game_language']
deadly_weapon_recoginizer = DeadlyWeaponRecoginizer(lang)
votes = {}
for image in images:
weapon_id = deadly_weapon_recoginizer.predict(image)
votes[weapon_id] = votes.get(weapon_id, 0) + 1
best_result = (None, 0)
for weapon_id in votes.keys():
if votes[weapon_id] > best_result[1]:
best_result = (weapon_id, votes[weapon_id])
response_payload = {
'status': 'ok',
'deadly_weapon': best_result[0],
'score': best_result[1],
'total': len(images),
}
return response_payload
