can you please help me solve the error.

Dear Besbes,
Firstly, I thank for your sharing code. It is very nice. I test with sagittal dataset with anterior cruciate ligament tear. However, I could not reproduce your results like train & valid AUC scores. So could you explain to clarify the issue? Or maybe I acquire a mistake somewhere?
Best regard. Linh

Hello, thanks for your implementation. However, I found that the "probs" for binary classification doesn't sum up to be 1.0.

        prediction = model.forward(image.float())
        loss = torch.nn.BCEWithLogitsLoss(weight=weight)(prediction, label)
        loss.backward()
        optimizer.step()

        loss_value = loss.item()
        losses.append(loss_value)

        probas = torch.sigmoid(prediction)

        y_trues.append(int(label[0][1]))
        y_preds.append(probas[0][1].item())

The issue mentioned is located in "https://github.com/ahmedbesbes/mrnet/blob/master/train.py"

Hi, thank you for your wonderful work. I'm trying the third part: Interpret Models' Predictions. I find I can't visualize some of the predictions. As follows:

IndexError Traceback (most recent call last)
in
1 import shutil
----> 2 create_patiens_cam(acl[0],plane)

in create_patiens_cam(case, plane)
19 logit = mrnet(mri)
20 size_upsample = (256, 256)
---> 21 feature_conv = feature_blobs[0]
22 # print(features_blobs[0])
23

IndexError: list index out of range

and then I did this:
mrnet = torch.load(f'../models/{model_name}')
mrnet = mrnet.to(device)
mrnet.eval()

hook the feature extractor

features_blobs = []
def hook_feature(module, input, output):
features_blobs.append(output.data.cpu().numpy())
mrnet._modules.get('pooling_layer').register_forward_hook(hook_feature);

...

global features_blobs
mri = mri.to(device)
logit = mrnet(mri)
size_upsample = (256, 256)
feature_conv = features_blobs[0]
h_x = F.softmax(logit, dim=1).data.squeeze(0)
probs, idx = h_x.sort(0, True)
probs = probs.cpu().numpy()
idx = idx.cpu().numpy()
slice_cams = returnCAM(features_blobs[-1], weight_softmax, idx[:1])

...

import shutil
create_patiens_cam(acl[0],plane)

Could you tell me what went wrong? thank you