Cross-dimension Affinity Distillation for 3D EM Neuron Segmentation

Accepted by CVPR-2024

Xiaoyu Liu, Miaomiao Cai, Yinda Chen, Yueyi Zhang, Te Shi, Ruobing Zhang, Xuejin Chen, Zhiwei Xiong*

University of Science and Technology of China

*Corresponding Author

Enviromenent

This code was tested with Pytorch 1.0.1 (later versions may work), CUDA 9.0, Python 3.7.4 and Ubuntu 16.04.

If you have a Docker environment, we strongly recommend you to pull our image as follows：

docker pull registry.cn-hangzhou.aliyuncs.com/renwu527/auto-emseg:v3.1

We release the Wafer4 dataset in Google Driver.

If you have any problem with the released code and dataset, please contact me by email (liuxyu@mail.ustc.edu.cn).