How to change __name__ = 'iou_score' and use it train.run file

[ramchandracheke]

Hi,
I am trying to create a IOU score per class and trying to change name = 'iou_score' according to class level. Can anyone knows how to modify this to get IoU score per class_label.
Thanks,

[ramchandracheke]

Hi,

I found solution as, Once the instance of train_epoch is created then we can change name as per below mentioned code.

metrics = [smp.utils.metrics.IoU(threshold=0.5, ignore_channels=[1,2,3]),
              smp.utils.metrics.IoU(threshold=0.5, ignore_channels=[0,2,3]),
              smp.utils.metrics.IoU(threshold=0.5, ignore_channels=[0,1,3])
]

train_epoch = smp.utils.train.TrainEpoch(
    model, 
    loss=loss, 
    metrics=metrics, 
    optimizer=optimizer,
    device=DEVICE,
    verbose=True,
)

train_epoch.metrics[0].__name__="IoU_Class0"
train_epoch.metrics[1].__name__="IoU_Class1"
train_epoch.metrics[2].__name__="IoU_Class2"

This will be helpful to create IoU score at class level.