PyTorch Implementation of U-ResNet used for track/shower pixel-labeling
ROOT
: data analysis framework. Defines file format, provides python bindings for our codeLArCV
: either version 1 and 2pytorch
: network implementationtensorboardX
: interface to log data that can be plotted with Tensorboardtensorflow-tensorboard
: (from tensorflow) for plotting loss/accuracyjupyter 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 networkplotlarcv1.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.