Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

The error message "Expected 5-dimensional input for 5-dimensional weight [8, 3, 3, 3, 3], but got 4-dimensional input of size [4, 384, 128, 48] instead" indicates that there is a mismatch between the expected input dimensions (5-dimensional) and the actual input dimensions provided (4-dimensional). #34

Open
Wuchen057 opened this issue Feb 21, 2024 · 2 comments

Comments

@Wuchen057
Copy link

After replacing the getitem method in line 77 of the file lib->medloaders>mrbrains2018.py with the following code, the error disappears and the training can proceed normally.

def __getitem__(self, index):
    t1_path, ir_path, flair_path, seg_path = self.list[index]
    t1_data, ir_data, flair_data, seg_data = np.load(t1_path), np.load(ir_path), np.load(flair_path), np.load(
        seg_path)

    return torch.FloatTensor(t1_data.copy()).unsqueeze(0), torch.FloatTensor(ir_data.copy()).unsqueeze(0), \
        torch.FloatTensor(flair_data.copy()).unsqueeze(0), torch.FloatTensor(seg_data.copy())

20240221150317
20240221150516

@littlewhite132
Copy link

非常感谢!
it really helps

@littlewhite132
Copy link

Hello, after correcting this mistake, I successfully trained the MRBrainS model. However, when I ran inference, I encountered an error: FileNotFoundError: [Errno 2] No such file or directory: './datasets/mrbrains_2018/training/mrbrains_2018-classes-4-list-train-samples-1.txt'. My training data was downloaded from the official website. How can I obtain this txt file?

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants