sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-17 06:25:25 +08:00
Code Activity

Upd: threshold for predict_valid in campaign_bonus

Browse Source
This commit is contained in:
sui-feng-cb
2025-12-04 19:23:50 +08:00
parent fa44b728bc
commit 7bd0413610
3 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
module/statistics/drop_statistics.py
View File

@@ -120,7 +120,7 @@ class DropStatistics:
if self.campaign_bonus.appear_on(image, similarity=similarity):
for button in [CAMPAIGN_BONUS_SINGLE, CAMPAIGN_BONUS]:
self.campaign_bonus.bonus_button = button
for item in self.campaign_bonus.stats_get_items(image):
for item in self.campaign_bonus.stats_get_items(image, mode='known'):
yield [ts, campaign, enemy_name, button.name, item.name, item.amount]
else:
raise ImageError('No campaign bonus on image.')
@@ -135,7 +135,7 @@ class DropStatistics:
print('')
logger.hr(f'Extract templates from {campaign}', level=1)
self.check_server(campaign)
for ts, file in tqdm(load_folder(self.drop_folder(campaign)).items()):
for ts, file in tqdm(load_folder(self.drop_folder(campaign), ext=['.png', '.jpg']).items()):
try:
self.parse_template(file)
except ImageError as e:
@@ -160,7 +160,7 @@ class DropStatistics:
with open(self.csv_file, 'a', newline='', encoding=DropStatistics.CSV_ENCODING) as csv_file:
writer = csv.writer(csv_file)
for ts, file in tqdm(load_folder(self.drop_folder(campaign)).items()):
for ts, file in tqdm(load_folder(self.drop_folder(campaign), ext=['.png', '.jpg']).items()):
try:
rows = list(self.parse_drop(file))
writer.writerows(rows)