FRA Grade Crossing Trespass Detection Machine Learning Model Development

This repository is home to the python scripts used to modify and fine-tune the pre-trained detection model.

Folder Structure

/PhaseOne Phase one model development using Tensorflow 1.x

/PhaseTwo Phase two model development using PyTorch 1.x

Requirements

• Python 3.x (Tested on 3.7.x)
• Tensorflow 1.15.x (Phase One)
• PyTorch >=1.5.x (Phase Two)

License

MIT