pytorch-uresnet

PyTorch Implementation of U-ResNet used for track/shower pixel-labeling

Dependencies

ROOT: data analysis framework. Defines file format, provides python bindings for our code
LArCV: either version 1 and 2
pytorch: network implementation
tensorboardX: interface to log data that can be plotted with Tensorboard
tensorflow-tensorboard: (from tensorflow) for plotting loss/accuracy
jupyter notebook: for testing code and inspecting images

ubuntu 16.10, ROOT 6, python 2.7.12, tensorflow-tensorboard (from tensorflow 1.4.1), cuda 8.0

uresnet.py: module that defines the network
plotlarcv1.ipynb: notebook used to test file IO commands. For LArCV classic files.
test.py.ipynb: notebook used to test file IO commands. For LArCV2 files.
train_wlarcv1.py: script for training.

Name		Name	Last commit message	Last commit date
Latest commit History 27 Commits
.gitignore		.gitignore
README.md		README.md
caffe_uresnet.py		caffe_uresnet.py
finetune_wlarcv1_cosmics.py		finetune_wlarcv1_cosmics.py
plotlarcv1.ipynb		plotlarcv1.ipynb
process_larcv1_caffe_uresnet.py		process_larcv1_caffe_uresnet.py
run_finetuning_job.sh		run_finetuning_job.sh
setup.sh		setup.sh
setup_container.sh		setup_container.sh
start_container.sh		start_container.sh
submit_finetuning.sh		submit_finetuning.sh
test.py.ipynb		test.py.ipynb
test_uresnet.py		test_uresnet.py
test_urestnet_larcv1.ipynb		test_urestnet_larcv1.ipynb
test_urestnet_larcv1_cosmic.ipynb		test_urestnet_larcv1_cosmic.ipynb
test_urestnet_larcv1_data.ipynb		test_urestnet_larcv1_data.ipynb
train_wlarcv1.py		train_wlarcv1.py
uresnet.py		uresnet.py