search

AllenCellModeling / pytorch_fnet

Three dimensional cross-modal image inference
Other
151 stars 61 forks source link

make pre prediction cropping, and post prediction pad with zero #66

Closed fcollman closed 6 years ago

fcollman commented 6 years ago

make test time dataset that has cropping and padding options that make the prediction the same size as the output.

test time model input modification options... pad/crop width and height parameters (can be negative to crop, positive to pad), default is to crop down to the nearest 16 option. at test time output .. reverse operation of test time modifications.

modify dataset object to have these crop options/choices recorded, modify dataset class to have function to reverse this operation.

needs to be done before #62

counkomol commented 6 years ago

Addressed with #68.

Instead of modifying the Dataset class, a cropper/padder object stores and can "undo" its last action.