WSMIS
WSMIS copied to clipboard
Question about IRN implementation
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?
I have the same problem. Have you solved it?