INNER CODE UNIT · Python
parse_raw_data
idealo/image-quality-assessment · data/TID2013/get_labels.py:36
def parse_raw_data(df):
samples = []
for i, row in df.iterrows():
max_entropy_dist = get_max_entropy_distribution(row['mos'])
samples.append({'image_id': row['id'].split('.')[0], 'label': max_entropy_dist.tolist()})
return samples
def main(target_file, source_file_mean):
df = get_dataframe(source_file_mean)
samples = parse_raw_data(df)
save_json(samples, target_file)
print('Done! Saved JSON at {}'.format(target_file))
if __name__ == '__main__':
parser = argparse.ArgumentParser()