Deep Neural Networks with Remote Sensing Data
This repo serves as the source code for the Final Project in Supervised Learning. This code will allow for GPU training of ImageNet based Deep Neural Networks, and is intented to be run on open source satellite imagery datasets.
- Build the container
docker build . -t dnn-rs
- Run the container
docker run -it dnn-rs
- Run the Code
python {train/eval}.py --help
In order to use a GPU, you will to install nvidia-docker2, and specify runtime=nvidia
in your docker run
command.