https://cellpose.readthedocs.io/en/latest/train.html from https://github.com/MouseLand/cellpose/issues/128#issuecomment-774732299