INNER CODE UNIT · Python
get_dataframe
idealo/image-quality-assessment · data/TID2013/get_labels.py:30
def get_dataframe(mean_raw_file):
df = pd.read_csv(mean_raw_file, header=None, sep=' ')
df.columns = ['mos', 'id']
return df
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)