innocuous-symmetry/vgg16-image-recognition
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

to do: debug existing JSON file feature

Browse Source
This commit is contained in:
Mikayla Dobson
2023-01-07 19:53:23 -06:00
parent c4e7e1301c
commit b1ba613468
3 changed files with 91 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
formatresult.py
View File

@@ -1,47 +1,74 @@
import json, os, shutil
from config import Config
def format_result(src_path, json_path):
def format_result(app_config: Config, json_path):
# dictionary to hold and later display our results
insertions_by_label = {}
data_path = app_config.data_path
# if pg_config is not None, run the postgres prediction[0] of this code
pg_config = app_config.pg_config
# if this is True, run the prediction[0] "for line in contents:" below
sort_by_match_strength = app_config.sort_by_match_strength
weak_results = 0
total_count = 0
# store analysis results in contents
with open(json_path) as results:
contents = json.load(results)
for qualifier in ['strong', 'moderate', 'fair', 'weak']:
if not os.path.exists("./predictions/" + qualifier):
os.makedirs('./predictions/' + qualifier)
# prepare individual locations for match strengths if sort is enabled
if sort_by_match_strength:
for qualifier in ['strong', 'moderate', 'fair', 'weak']:
if not os.path.exists("./predictions/" + qualifier):
os.makedirs('./predictions/' + qualifier)
# handles data for each photo
for line in contents:
img_path = src_path + line['path']
img_path = data_path + line['path']
prediction = line['prediction']
for section in prediction:
total_count += 1
guess_label = section[0][1]
match_strength = 'weak/'
if float(section[0][2]) > 0.9:
match_strength = 'strong/'
elif float(section[0][2]) > 0.75:
match_strength = 'moderate/'
elif float(section[0][2]) > 0.5:
match_strength = 'fair/'
elif match_strength == 'weak/':
weak_results += 1
# handles data for first prediction for a given photo
total_count += 1
guess_label = prediction[0][0][1]
match_strength = 'weak'
if not guess_label in insertions_by_label:
insertions_by_label[guess_label] = 0
# assign value for match strength
if float(prediction[0][0][2]) > 0.9:
match_strength = 'strong'
elif float(prediction[0][0][2]) > 0.7:
match_strength = 'moderate'
elif float(prediction[0][0][2]) > 0.4:
match_strength = 'fair'
elif match_strength == 'weak':
weak_results += 1
# modify variable for path
match_strength = match_strength + "/"
if not guess_label in insertions_by_label:
insertions_by_label[guess_label] = 0
# copy file to appropriate location, depending on if sorting
if sort_by_match_strength:
if (not os.path.exists("./predictions/" + match_strength + guess_label)):
os.makedirs("./predictions/" + match_strength + guess_label)
if (not os.path.exists('./predictions/' + match_strength + guess_label + '/' + img_path)):
shutil.copy(img_path, "./predictions/" + match_strength + guess_label)
insertions_by_label[guess_label] = insertions_by_label[guess_label] + 1
else:
if (not os.path.exists("./predictions/" + guess_label)):
os.makedirs("./predictions/" + guess_label)
if (not os.path.exists('./predictions/' + guess_label + '/' + img_path)):
shutil.copy(img_path, "./predictions/" + guess_label)
insertions_by_label[guess_label] = insertions_by_label[guess_label] + 1
print(str(weak_results) + " weak result(s) of a total " + str(total_count) + " input(s)\n")
print("By subject:\n\n")
print("By subject:\n")
for k, v in insertions_by_label.items():
print(k + ": " + str(v) + " file(s) found")