文件名称:DSI-Net:DSI网
文件大小:438KB
文件格式:ZIP
更新时间:2024-05-09 05:34:35
Python
DSI网 该存储库是论文“ DSI-Net:用于通过内窥镜图像进行联合分类和细分的深度协同交互网络”的正式PyTorch实施,正在TMI 2021进行审查。 依存关系 Python 3.6 PyTorch> = 1.3.0 麻木 顶尖 斯克莱恩 matplotlib 皮尔 用法 培训DSI-Net python train_DSI_Net . py - - gpus 0 - - K 100 - - alpha 0.05 - - image_list 'data/WCE/WCE_Dataset_image_list.pkl' 测试DSI-Net 下载我们训练有素的模型,并将其放在\checkpoints 。 测试前请裁切黑边。 python test_DSI_Net . py - - img_path xx - - model xxx 接触 ( )
【文件预览】:
DSI-Net-main
----logs()
--------README.md(1B)
----data()
--------pre_model()
----visualization()
--------utils.py(1KB)
----dataset()
--------my_datasets.py(3KB)
--------transform_customize.py(1KB)
----net()
--------sync_batchnorm()
--------loss.py(2KB)
--------models.py(5KB)
--------__pycache__()
--------gumbel.py(824B)
--------xception.py(8KB)
--------modules.py(11KB)
--------convs.py(526B)
--------ASPP.py(2KB)
----checkpoints()
--------README.md(1B)
----Figs()
--------Framework.png(395KB)
----training_script.sh(388B)
----config.py(586B)
----README.md(835B)
----utils()
--------fp16util.py(1KB)
--------logger.py(177B)
--------metrics.py(1KB)
--------meter.py(611B)
----train_DSI_Net.py(11KB)