Data

Tools

Indexes

Applications

Experiments

Open Source Science

stc-unet

https://github.com/ahuweia/stc-unet

Science Score: 26.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ○
    CITATION.cff file
  • ✓
    codemeta.json file
    Found codemeta.json file
  • ✓
    .zenodo.json file
    Found .zenodo.json file
  • ○
    DOI references
  • ○
    Academic publication links
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (6.6%) to scientific vocabulary
Last synced: 10 months ago · JSON representation

Repository

Basic Info
  • Host: GitHub
  • Owner: ahuweia
  • License: apache-2.0
  • Language: Python
  • Default Branch: main
  • Size: 10.3 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 1
  • Releases: 0
Created over 2 years ago · Last pushed over 2 years ago
Metadata Files
Readme License Citation

README.md

STC-UNet

This study is based on mmsegmentation.

1 Installation

Important versions

python==3.8 pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 mmcv-full==1.7.0 timm==0.4.12 segmentation-models-pytorch==0.2.0 yapf==0.40.1

Step 1. Create a conda environment and activate it.

conda create --name py38 python=3.8 -y conda activate py38

Step 2. Install PyTorch using following command.

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

Step 3. Install MMCV using following command.

pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10/index.html

Step 4. Install other packages.

pip install mmengine tensorboard timm==0.4.12 segmentation-models-pytorch==0.2.0 opencv-python einops yapf==0.40.1

Step 5. Clone this repository.

git clone https://github.com/ahuweia/STC-UNet.git cd STC-UNet pip install -e .

2 Datasets

This study uses the KiTS19 dataset for related experiments.

3 Training

python tools/train.py my_config/STC-UNet.py

4 Inference

python tools/test.py "the path of your test data" "my_config/STC-UNet.py" "path/to/checkpoint.pth"

Owner

  • Name: HuWei
  • Login: ahuweia
  • Kind: user

GitHub Events

Total
  • Issues event: 1
  • Issue comment event: 2
Last Year
  • Issues event: 1
  • Issue comment event: 2

Dependencies

docker/Dockerfile docker
  • pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
docker/serve/Dockerfile docker
  • pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
mmsegmentation.egg-info/requires.txt pypi
  • cityscapesscripts *
  • codecov *
  • flake8 *
  • interrogate *
  • matplotlib *
  • mmcls >=0.20.1
  • mmcv-full <1.7.0,>=1.4.4
  • numpy *
  • packaging *
  • prettytable *
  • pytest *
  • xdoctest >=0.10.0
  • yapf *
requirements/docs.txt pypi
  • docutils ==0.16.0
  • myst-parser *
  • sphinx ==4.0.2
  • sphinx_copybutton *
  • sphinx_markdown_tables *
requirements/mminstall.txt pypi
  • mmcls >=0.20.1
  • mmcv-full >=1.4.4,<1.7.0
requirements/optional.txt pypi
  • cityscapesscripts *
requirements/readthedocs.txt pypi
  • mmcv *
  • prettytable *
  • torch *
  • torchvision *
requirements/runtime.txt pypi
  • matplotlib *
  • mmcls >=0.20.1
  • numpy *
  • packaging *
  • prettytable *
requirements/tests.txt pypi
  • codecov * test
  • flake8 * test
  • interrogate * test
  • pytest * test
  • xdoctest >=0.10.0 test
  • yapf * test
requirements.txt pypi
setup.py pypi