cellpose v0.6
Pytorch is now the default deep neural network software for cellpose. Mxnet will still be supported. To install mxnet (CPU), run pip install mxnet-mkl
. To use mxnet in a notebook, declare torch=False
when creating a model, e.g. model = models.Cellpose(torch=False)
. To use mxnet on the command line, add the flag --mxnet
, e.g. python -m cellpose --dir ~/images/ --mxnet
. The pytorch implementation is 20% faster than the mxnet implementation when running on the GPU and 20% slower when running on the CPU.
Dynamics are computed using bilinear interpolation by default instead of nearest neighbor interpolation. Set interp=False
in model.eval
to turn off. The bilinear interpolation will be slightly slower on the CPU, but it is faster than nearest neighbor if using torch and the GPU is enabled.