search

mattmacy / vnet.pytorch

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation
https://mattmacy.github.io/vnet.pytorch
BSD 3-Clause "New" or "Revised" License
688 stars 201 forks source link

Prediction script #7

Closed kiranvaidhya closed 7 years ago

kiranvaidhya commented 7 years ago

Can you provide a prediction script for iterating and predicting through 3D volumes?

kiranvaidhya commented 7 years ago

The issue is referenced here

mattmacy commented 7 years ago

Passing --inference with the model path worked last time I tried it. See inference() in train.py.