AllenCellModeling / pytorch_fnet

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

General updates #98

Closed counkomol closed 6 years ago

counkomol commented 6 years ago

Key changes: • Fixes bug in BufferedPatchDataset where patches weren't being sampled from the entire range of the buffered images. (Also, a test was added to check for this.) • Adds training option to periodically test the model on a validation set. • Updates model load function so the load path can be either a directory or some arbitrary saved model checkpoint. • Adds alternative "fnet_model" for predicting 3d volumes using a 2d network.