41118_SortBot

An intelligent robotic sorting system built in PyBullet that combines deep reinforcement learning (PPO) with object detection (YOLOv8). The uArm robot detects, picks up, and sorts coloured boxes into their designated zones autonomously.

---

Team Composition

Name	Role
Minh K. Nguyen	Simulation, Robot control, PPO, YOLO, Vision, integration & deployment
Lauren Seeto	Robot control, Robot Modelling, PPO
Michele Liang	Vision, YOLO, documentation
Luis Pratama	Vision, YOLO video director, voice, editor
Matthew Truong	Simulation Modelling

This project was developed as part of the 41118 Artificial Intelligence in Robotics unit at UTS. Special thanks to Dr. Raphael Falque for his guidance and teaching throughout this subject.

---

Installation Instructions

Clone the repository

HTTPS:

git clone https://github.com/FattDuckk/ai_project.git

ssh:

git clone [email protected]:FattDuckk/ai_project.git

Running

python3 main.py

Dependencies

Install PPO

pip install stable_baselines3

Install YOLOv8

pip install ultralytics

Install PyBullet

pip install pybullet

Install Numpy

pip install numpy

Install Gym

pip install gym