Skip to content

Latest commit

 

History

History
6 lines (5 loc) · 1019 Bytes

Nets.md

File metadata and controls

6 lines (5 loc) · 1019 Bytes

Neural network scripts

The TrailNet DNN is based on the popular ResNet-18 architecture with a few modifications (refer to the paper for more details). The caffe_srelu_resnet.py script in ResNet directory generates a prototxt file (srelu-resnet-18.prototxt) which is a description of the DNN architecture that is used at the first stage of training using the original Trails dataset. At the second stage of training, the network is modified and a second "head" is added which is then trained on the translation dataset. Only the last layer of this second head is trained; all other layers (convolutional etc) are frozen. The TrailNet_SResNet-18.prototxt file contains a network architecture for the second step. During training, both networks use custom data augmentation techniques as well as a custom loss function. This functionality is implemented as Caffe Python layers in the python-layers.py file. This file needs to be specified in DIGITS when launching a job.