DNN-RS

Deep Neural Networks with Remote Sensing Data

Description

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.

Running the Code

1. Build the container

docker build . -t dnn-rs

2. Run the container

docker run -it dnn-rs

3. Run the Code

python {train/eval}.py --help

GPU Usage

In order to use a GPU, you will to install nvidia-docker2, and specify runtime=nvidia in your docker run command.