GitHub - ginobilinie/Strainet: The work I have done for pelvic organ segmentation, implemented by pytorch.

This is a Pytorch 2D version of the STRAINet. Of course, I have a lot of new functions implemented besides the ones I mention in TNNLS. As I have deleted some sensitve new functions, the codes may have some bugs in some cases. If you have any problems about the code, please contact dongnie@cs.unc.edu

The main entrance is runGAN.py.

I'll upload explanations in later days.

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
PSPNet.py		PSPNet.py
README.md		README.md
blocks.py		blocks.py
computeMetric4Subjects.py		computeMetric4Subjects.py
config.py		config.py
convertMedImgFormat.py		convertMedImgFormat.py
dataClean.py		dataClean.py
dataset.py		dataset.py
deeplab_resnet.py		deeplab_resnet.py
deform_cnn.py		deform_cnn.py
deform_conv.py		deform_conv.py
deform_conv_layer.py		deform_conv_layer.py
drawContour.m		drawContour.m
ganComponents.py		ganComponents.py
gradientCheckExample.py		gradientCheckExample.py
mha_read_header.m		mha_read_header.m
mha_read_volume.m		mha_read_volume.m
nnBuildUnits.py		nnBuildUnits.py
refinenet_4cascade.py		refinenet_4cascade.py
runDeepLabV2.py		runDeepLabV2.py
runGAN.py		runGAN.py
runPSPNet.py		runPSPNet.py
runRefineNet.py		runRefineNet.py
runTesting.py		runTesting.py
runVisualization.py		runVisualization.py
srresnet.py		srresnet.py
test.py		test.py
utils.py		utils.py
visualize.py		visualize.py

ginobilinie/Strainet

Folders and files

Latest commit

History

Repository files navigation

About

Resources

Stars

Watchers

Forks

Languages