SegFormer3D icon indicating copy to clipboard operation
SegFormer3D copied to clipboard

how to train my own dataset?

Open wydilearn opened this issue 1 year ago • 7 comments

I can not preprocess the dataset using providing codes. Please provide full explicit comments for how to use other datset.

wydilearn avatar Aug 26 '24 11:08 wydilearn

My terminal only print the two sentences without doing anything: "started preprocessing Brats2017..." "finished preprocessing Brats2017..." Is this a incomplete codes?

wydilearn avatar Aug 26 '24 12:08 wydilearn

The preprocessing code only processes Brats dataset. What is your dataset ?

bnavard avatar Aug 26 '24 17:08 bnavard

My dataset has only one channel, and I have transform my dataset correctly, but I still cannot train the dataset. Could you please explain which part of the codes that I need to edit?

wydilearn avatar Aug 27 '24 03:08 wydilearn

In addition, my labels have three classes.

wydilearn avatar Aug 27 '24 03:08 wydilearn

And my image size is 128x48x96

wydilearn avatar Aug 27 '24 03:08 wydilearn

无标题 In addition, I don't find the test codes. I can not test the model after training it.

wydilearn avatar Aug 27 '24 09:08 wydilearn

If your dataset is BraTS you can use our datapreprcoessing. What is your dataset ? If you are using a data with different shape (other than 128x128x128) you need to change the SegFormer3D architecture config such that the path embeddings works with your dataset size. For that you should change your experiment config.

bnavard avatar Aug 28 '24 13:08 bnavard

If your dataset is BraTS you can use our datapreprcoessing. What is your dataset ? If you are using a data with different shape (other than 128x128x128) you need to change the SegFormer3D architecture config such that the path embeddings works with your dataset size. For that you should change your experiment config.

hello @bnavard , thanks for creating and maintaining this repo!

for e.g. could you suggest what changes should be made for a dataset with shape (D, X, Y)=(32, 128, 128) ?

embed_dims: list = [32, 64, 160, 256],
patch_kernel_size: list = [7, 3, 3, 3],
patch_stride: list = [4, 2, 2, 2],
patch_padding: list = [3, 1, 1, 1],

If I understand correct the above lists correspond to i-th MixViT block operations uniform for x,y,z dims

EDIT: have been resolved. The comments here for handling non-symmetric input helped a lot issues #6

i-mein avatar Oct 18 '24 08:10 i-mein