Open shivahanifi opened 1 day ago
# Prepare class-wise IoU for logging
class_wise_IU = evaluation_result['class_wise_IU']
class_iou_dict = {f"class_{i}": iou for i, iou in enumerate(class_wise_IU)}
# Log the class-wise IoU as a dictionary
wandb.log({"class_wise_IU": class_iou_dict})
# Prepare class-wise IoU for logging
class_wise_IU = evaluation_result['class_wise_IU']
label_desc = df['Desc'].tolist()
wandb.config.labels_Desc = label_desc
class_iou_dict = {f"class_{i}:{label_desc[i]}": iou for i, iou in enumerate(class_wise_IU)}
# Log evaluation results
wandb.log({"frequency_weighted_IU": evaluation_result['frequency_weighted_IU'],
"mean_IU": evaluation_result['mean_IU'],
"class_wise_IU": class_iou_dict,
})
This is the 'class_wise_IU'. I want to modify the its log in wandb to have it visually tracable.