CS230_DAA

Code for CS230 course project on implementing a vision-based detect and avoid (DAA) system.

Getting set up with X-Plane 11

The data generation files in this repository require X-Plane 11 and the XPlaneConnect Plugin. Information about how to set this up can be found here.

File Overview

src/data_generation/generate_traffic_data_py: script to generate images of intruder aircraft at various relative positions and save to a specified folder

src/data_generation/label_traffic_data.py: script to generate bounding box labels for a set of images given the relative position of the intruder aircraft

src/data_generation/tune_bounding.ipynb: notebook used to tune parameters for bounding box labeling

src/data_generation/xpc3.py: code required to interface with X-Plane 11

src/yolo_v5/: code to train yolo network (see here)

src/util.py: utility functions for loading in networks and evaluating them on images from X-Plane 11

src/CS230_results.ipynb: notebook with results analysis and figure generation codes

models/: contains models for three trials

results/: gifs of results