The 2D data，can I complete training and testing on just one plane?

twni2016 / OrganSegRSTN_PyTorch

PyTorch implementation of OrganSegRSTN - CVPR 2018

http://ccvl.jhu.edu/

MIT License

107 stars 33 forks source link

The 2D data，can I complete training and testing on just one plane? #11

Closed lsjAI closed 2 years ago

lsjAI commented 3 years ago

My data is 2d, Is it possible to complete training and testing on only one plane? Thank you~

twni2016 commented 2 years ago

We provide CURRENT_PLANE argument to train and test on one plane. But I guess you need to reshape your 2D data to 3D with an extra axis if you want to directly use the code.