This repository contains the complete code and data for my EE695 FYP: "Design and Implementation of an Autonomous Connect Four Playing Robot Arm"
- computer_vision: code relating to computer vision testing of the system.
- colour_segmentation: colour segmentation method testing and source code. Contains the connect4_cv package which provides the colour segmentation detector and various useful tools.
- yolo: yolo method testing. Contains scripts for dataset collection and manipulation. Model weights are also present.
- gameplay_simulator: source code for a connect four gameplay simulator.
- ros_system: source code for the ROS1 based integrated system.
- testing_data: data from testing of the system.