【文件属性】:
文件名称:thyroid_segmentation:使用UNets和变体的超声甲状腺结节分割
文件大小:150KB
文件格式:ZIP
更新时间:2021-05-18 22:22:15
computer-vision segmentation unet Python
甲状腺结节分割
该存储库包含用于在超声图像中分割甲状腺结节的代码和模型。 使用的数据集:
安装
主要代码以名为“ tnseg”的Python包编写。 将此存储库克隆到您的计算机后,请使用以下命令安装
cd cloned/path
pip install .
然后,您应该能够在Python中使用该软件包:
import matplotlib . pyplot as plt
from tnseg import dataset , models , loss , opts , evaluate
运行模型
用于模型训练和评估的脚本位于/ scripts /下。
python -u scripts/train.py config_files/defaults.config
运行模型时,输出将保存在“ outputs /”文件夹中,并保存在以实验名称命名的文件夹中(应在配置文件中指定)。
【文件预览】:
thyroid_segmentation-master
----notebooks()
--------.ipynb_checkpoints()
----tnseg()
--------Augmentor()
--------models()
--------opts.py(8KB)
--------loss.py(1KB)
--------dataset.py(10KB)
--------ufarray.py(5KB)
--------__init__.py(19B)
--------evaluate.py(7KB)
----scripts()
--------train.py(10KB)
--------config_files()
----requirements.txt(15B)
----LICENSE(1KB)
----setup.py(441B)
----README.md(2KB)
----outputs()
--------exp_1()
----.gitignore(60B)