Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

Use custom models and custom data loading functions #2646

Open
sty21211250 opened this issue Dec 17, 2024 · 0 comments
Open

Use custom models and custom data loading functions #2646

sty21211250 opened this issue Dec 17, 2024 · 0 comments
Assignees

Comments

@sty21211250
Copy link

sty21211250 commented Dec 17, 2024

Hello, I have now written a custom model that needs to enter two data at once, such as
def forward(self, x_current, x_previous):
I need to modify the data_loader function again, but I see that the file of nnUNetDataset says that this is a read-only file, so I would like to ask which part I should modify to meet my requirements

And I certainly thought about just making a simple change to the nnUNetDataLoader3D function or not, transforming the tensor passed in to the model to get a new data. The model I defined takes into account the time factor, so I need to input data of two time points at the same time. If I want to transform the incoming tensor to obtain a new data, I should use the registration field to transform it. However, due to the influence of data enhancement operation, the deformation field may play a counterproductive role, so I am a little confused now.

Mainly because the pre-processing and post-processing of nnunet are too excellent, and the related auxiliary functions are also very perfect, which can greatly facilitate my task. If you can point out the direction, I would appreciate it very much, and then I will attach my model code

[.docx](https://github.com/user-attachments/files/18163876/default.docx)
`
this is my network

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

No branches or pull requests

2 participants