Question about IRN implementation

[sgadgil6]

While loading the dataset for training IRNet (PneumothoraxAffinityDataset) in /irn/voc2/dataloader.py, you load the mask on line 324. Is this the same segmentation mask saved on line 80 of cam.py in /inference/cam_generation/cam.py? If so, the original IRNet code had a step for converting cams to ir labels (irn/step/cam_to_ir_label.py). Do you make use of that step for the pneumothorax dataset or do you directly use the segmentation mask for training IRNet?

[Mugen-chen]

I have the same problem. Have you solved it？