ImgX-DiffSeg
ImgX-DiffSeg copied to clipboard
A JAX-based deep learning framework for image segmentation using diffusion models.
# Description As mentioned in #18 and #20, the usage of TFDS may not be obvious. While the [inference example](https://github.com/mathpluscode/ImgX-DiffSeg/blob/main/examples/segmentation/inference.ipynb) demos how to use a trained network on custom data,...
您好,非常感谢您的精彩的工作。但是在运行过程中,出现了一些问题。我的系统是Ubuntu,按照您的教程配置环境,并在amos_ct数据集上进行训练,其他似乎都没有问题,但是在运行imgx_train data=amos_ct task=gaussian_diff_seg时,运行似乎卡住,在1个小时之后,仍然停留在这里。请问是什么原因呢?由于直接运行会出一些错误,我在按照教程配置完环境后,将scipy降低版本到1.10.1,并将./imgx/conf/config.yaml文件中的project: imgx entity: wandb_entity修改为project: 和entity: 。如上,另外,我观察我的wandb的workspace,结果如下图所示。非常期待您的回答与帮助,谢谢!  