taesiri's picture
Initial Commit
8c20033
import os
def get_label_for_query(image_url, model_name):
fourway_label = image_url.split('/')[-2]
if fourway_label=='both_correct':
return 'Correct'
if fourway_label=='both_wrong':
return 'Wrong'
if fourway_label == 'chm_correct_knn_incorrect' and model_name == 'CHM':
return 'Correct'
elif fourway_label == 'knn_correct_chm_incorrect' and model_name == 'KNN':
return 'Correct'
return 'Wrong'
def get_category(image_url):
return image_url.split('/')[-2]
def translate_winds_to_names(winds):
return [folder_to_name[x] for x in winds]