SlicerIGT / aigt

Deep learning software modules for image-guided medical procedures
http://www.slicerigt.org
BSD 3-Clause "New" or "Revised" License
61 stars 43 forks source link

Resizing images in prepare_data vs. resizing using Pytorch Dataset #36

Closed chriscyyeung closed 1 year ago

chriscyyeung commented 1 year ago

Seems redundant to have both. Might make more sense to resize using Pytorch transforms during training?

ungi commented 1 year ago

Some redundancy is not a problem, especially if we can easily skip one resizing step out of the two. Feel free to implement resize transform in the training script. If the input is already resized by the prepare_data script, then resizing will be automatically skipped during training, and training will require less time, data traffic, and storage space during training. It's a nice option to keep.