Confirmation Question: Input to Model?
RylanSchaeffer opened this issue · 3 comments
RylanSchaeffer commented
Hi! Quick question: what is the actual input to the network? Is the input the full 2D/3D image, or is the input smaller (perhaps random) sections of the full image?
I think the answer is the full image, but I want to confirm.
RylanSchaeffer commented
Ok maybe input images are cropped first, based on this line
https://github.com/askerlee/segtran/blob/master/code/train3d.py#L528
Could you clarify?
askerlee commented
Correct. It's randomly cropped during training to reduce overfitting. Otherwise the model always sees the same image and easily remembers the correct mask.
RylanSchaeffer commented
Thank you!