/icsrl-deep-learning

Primary LanguagePython

Deep Learning Templates

This repository contains template code for the following deep learning tasks:

  1. Image Classification

  2. Object Detection

object_detection1 \ object_detection2 \ object_detection3

  1. Semantic Segmentation

semantic_seg1 \ semantic_seg2 \ semantic_seg3

  1. Reinforcement Learning (Proximal Policy Optimization)

flappy_bird \ breakout \

Getting Started

It is intended to run on our lab servers

  1. icsrl-exxact1.ece.gatech.edu
  2. icsrl-srv2.ece.gatech.edu

Lastly there are two things you need to configure before running:

  1. Dataset Path
  2. Transfer Learning weights

Both of these are available on the server and can be configured by setting the exxact flag in the script.

  • exxact=1 implies you are working on the exxact server
  • exxact=0 implies you are working on the icsrl2 server

Affiliation

Georgia Institute of Technology, ICSRL (http://icsrl.ece.gatech.edu/)

Alt text

License

This project is licensed under the MIT License - see the LICENSE.md file for details