ssTem Dataset : dataset/train-volume.tif,train-labels.tif,test-volume.tif
- clone repository
git clone https://github.com/realJun9u/unet_sstem.git
- Install Dependancy
# in keras or pytorch
pip install -r requirement.txt
- Setup Dataset (Seperate Datasets in ./dataset)
python setup.py
- Train and Evaluate Model
python train.py
- Predict Model (Create Prediction Images in result directory)
python predict.py
- Visualize Model
python visualize.py
- Analyze Result
tensorboard --logdir=logs --host={host} --port={port}
- CUDA 버전
nvcc -V
- cuDNN 버전
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2