NeodroidVision 0.2.0

Computer Vision algorithm implementations, intended for use with the Neodroid platform

Vision

This repository will host implementation computer vision algorithms applying the Neodroid platform.

---

Neodroid is developed with support from Research Council of Norway Grant #262900. (https://www.forskningsradet.no/prosjektbanken/#/project/NFR/262900)

---

Contents Of This Readme

• Algorithms
• Requirements
• Usage
• Results
  • Segmentation
• Contributing
• Other Components

Algorithms

• SAGAN - Generative model
• CVAE - Generative model
• BVAE - Generative model
• Vanilla-CNN - Classification
• Vanilla-MLP - Classification
• YOLO - Segmentation

Requirements

• pytorch
• tqdm
• Pillow
• numpy
• matplotlib
• torchvision
• torch
• Neodroid
• pynput

To install these use the command:

pip3 install -r requirements.txt

Usage

Export python path to the repo root so we can use the utilities module

export PYTHONPATH=/path-to-repo/

For training a agent use:

python3 procedures/train_agent.py

For testing a trained agent use:

python3 procedures/test_agent.py

Results

Segmentation

Code

Screenshots

Contributing

See guidelines for contributing here.

Citation

For citation you may use the following bibtex entry:

@misc{neodroid-vision,
  author = {Heider, Christian},
  title = {Neodroid Vision},
  year = {2019},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/aivclab/vision}},
}

Authors

• Christian Heider Nielsen - cnheider

Here other contributors to this project are listed.