您好，请问如果是分类良恶性肿瘤，训练集是nii图像，标签是csv文件可以训练吗

YubiaoYue / MedMamba

This is the official code repository for "MedMamba: Vision Mamba for Medical Image Classification"

297 stars 24 forks source link

Open 111678900 opened 5 months ago

111678900 commented 5 months ago

您好，请问如果是分类良恶性肿瘤，训练集是nii图像，标签是csv文件可以训练吗

YubiaoYue commented 5 months ago

您好，可以的。但是需要自己定义Dataset

111678900 commented 4 months ago

怎么定义呢请问，因为我看训练集只要了图像（）

111678900 commented 4 months ago

而且我看您的代码似乎不支持nii格式

YubiaoYue commented 4 months ago

您好，您需要自己利用pytorch重写dataset类。可以看看相关教程。此外，Medmamba目前只支持2D图像，如果您的图像不是2D，可能需要自己定义一个单独的Stem模块