├── .gitattributes ├── .gitignore ├── Dockerfile ├── Layers.py ├── OutputModule.py ├── README.md ├── VNet.py ├── VNetDiagram.png ├── attention.py ├── configs ├── config.json ├── config_docker.json └── config_sample.json ├── cxx ├── CMakeLists.txt ├── README.md ├── SafeQueue.h ├── ThreadPool.h ├── main.cxx ├── tf_inference.cpp └── tf_inference.h ├── data ├── evaluate │ └── 2 │ │ ├── image.nii │ │ └── label.nii ├── testing │ └── 1 │ │ ├── image.nii │ │ └── label.nii └── training │ └── 0 │ ├── image.nii │ └── label.nii ├── evaluate.py ├── layers2.py ├── main.py ├── meta_to_pb.py ├── model.py ├── networks.py ├── pipeline ├── NiftiDataset2D.py ├── NiftiDataset3D.py ├── __init__.py ├── pipeline.py ├── pipeline2D.yaml └── pipeline3D.yaml ├── requirements.txt ├── test.py ├── train.py └── utils ├── TensorRT └── freeze.py ├── batch_evaluate ├── README.md ├── batch_evaluate.py ├── batch_evaluate_step.py ├── batch_evaluate_stride.py └── main.py ├── bounding_box ├── README.MD ├── batch_bbox.py ├── bbox.png ├── bbox.py └── main.py └── prepare_data ├── README.md ├── adam_unzip.py ├── binarize.py ├── check_header_consistency.py ├── image_fit_label.py ├── image_partition.py ├── lits.py └── unify_header.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/Dockerfile -------------------------------------------------------------------------------- /Layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/Layers.py -------------------------------------------------------------------------------- /OutputModule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/OutputModule.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/README.md -------------------------------------------------------------------------------- /VNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/VNet.py -------------------------------------------------------------------------------- /VNetDiagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/VNetDiagram.png -------------------------------------------------------------------------------- /attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/attention.py -------------------------------------------------------------------------------- /configs/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/configs/config.json -------------------------------------------------------------------------------- /configs/config_docker.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/configs/config_docker.json -------------------------------------------------------------------------------- /configs/config_sample.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/configs/config_sample.json -------------------------------------------------------------------------------- /cxx/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/CMakeLists.txt -------------------------------------------------------------------------------- /cxx/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/README.md -------------------------------------------------------------------------------- /cxx/SafeQueue.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/SafeQueue.h -------------------------------------------------------------------------------- /cxx/ThreadPool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/ThreadPool.h -------------------------------------------------------------------------------- /cxx/main.cxx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/main.cxx -------------------------------------------------------------------------------- /cxx/tf_inference.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/tf_inference.cpp -------------------------------------------------------------------------------- /cxx/tf_inference.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/cxx/tf_inference.h -------------------------------------------------------------------------------- /data/evaluate/2/image.nii: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/data/evaluate/2/image.nii -------------------------------------------------------------------------------- /data/evaluate/2/label.nii: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/data/evaluate/2/label.nii -------------------------------------------------------------------------------- /data/testing/1/image.nii: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/data/testing/1/image.nii -------------------------------------------------------------------------------- /data/testing/1/label.nii: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/data/testing/1/label.nii -------------------------------------------------------------------------------- /data/training/0/image.nii: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/data/training/0/image.nii -------------------------------------------------------------------------------- /data/training/0/label.nii: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/data/training/0/label.nii -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/evaluate.py -------------------------------------------------------------------------------- /layers2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/layers2.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/main.py -------------------------------------------------------------------------------- /meta_to_pb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/meta_to_pb.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/model.py -------------------------------------------------------------------------------- /networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/networks.py -------------------------------------------------------------------------------- /pipeline/NiftiDataset2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/pipeline/NiftiDataset2D.py -------------------------------------------------------------------------------- /pipeline/NiftiDataset3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/pipeline/NiftiDataset3D.py -------------------------------------------------------------------------------- /pipeline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pipeline/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/pipeline/pipeline.py -------------------------------------------------------------------------------- /pipeline/pipeline2D.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/pipeline/pipeline2D.yaml -------------------------------------------------------------------------------- /pipeline/pipeline3D.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/pipeline/pipeline3D.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | SimpleITK 2 | tqdm 3 | protobuf==3.19.6 4 | PyYAML -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/train.py -------------------------------------------------------------------------------- /utils/TensorRT/freeze.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/batch_evaluate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/batch_evaluate/README.md -------------------------------------------------------------------------------- /utils/batch_evaluate/batch_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/batch_evaluate/batch_evaluate.py -------------------------------------------------------------------------------- /utils/batch_evaluate/batch_evaluate_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/batch_evaluate/batch_evaluate_step.py -------------------------------------------------------------------------------- /utils/batch_evaluate/batch_evaluate_stride.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/batch_evaluate/batch_evaluate_stride.py -------------------------------------------------------------------------------- /utils/batch_evaluate/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/batch_evaluate/main.py -------------------------------------------------------------------------------- /utils/bounding_box/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/bounding_box/README.MD -------------------------------------------------------------------------------- /utils/bounding_box/batch_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/bounding_box/batch_bbox.py -------------------------------------------------------------------------------- /utils/bounding_box/bbox.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/bounding_box/bbox.png -------------------------------------------------------------------------------- /utils/bounding_box/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/bounding_box/bbox.py -------------------------------------------------------------------------------- /utils/bounding_box/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/bounding_box/main.py -------------------------------------------------------------------------------- /utils/prepare_data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/README.md -------------------------------------------------------------------------------- /utils/prepare_data/adam_unzip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/adam_unzip.py -------------------------------------------------------------------------------- /utils/prepare_data/binarize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/binarize.py -------------------------------------------------------------------------------- /utils/prepare_data/check_header_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/check_header_consistency.py -------------------------------------------------------------------------------- /utils/prepare_data/image_fit_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/image_fit_label.py -------------------------------------------------------------------------------- /utils/prepare_data/image_partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/image_partition.py -------------------------------------------------------------------------------- /utils/prepare_data/lits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/lits.py -------------------------------------------------------------------------------- /utils/prepare_data/unify_header.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jackyko1991/vnet-tensorflow/HEAD/utils/prepare_data/unify_header.py --------------------------------------------------------------------------------