How to switch dataset? - Githubissues

mattmacy / vnet.pytorch

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

https://mattmacy.github.io/vnet.pytorch

BSD 3-Clause "New" or "Revised" License

693 stars 200 forks source link

How to switch dataset? #3

Closed tommy-qichang closed 7 years ago

tommy-qichang commented 7 years ago

Hi @mattmacy Thanks for your great work. I just curious how to migrate LUNA16 dataset into other private dataset? Any suggestion? Thanks.

mattmacy commented 7 years ago

Look at how the loaders are implemented. You'll need to write a loader for your dataset. The LUNA16 loader could in principle be used for anything that uses mhd files.

mattmacy commented 7 years ago

There's nothing particularly data set specific to this code LUNA16 just happens to be what I was using.