This repository contains the baseline code for the challenge titled “Drone Perspective Crowd Counting” from the GAIIChallenge hosted at HeyWhale. The objective is to develop an algorithm that accurately counts the number of people from a drone’s perspective using dual optical sensors (presumably RGB and thermal or depth cameras), addressing challenges such as varying densities, occlusions, and altitude-induced perspective changes.
model/
: Directory to store trained models. Created during setup to separate trained weights and configurations from source code.train.py
: Script responsible for training the crowd counting model using provided or custom datasets. It implements necessary data preprocessing, model architecture, and training loop.test.py
: Script designed to evaluate the trained model on a test dataset and generate output predictions, which are then redirected to ans.txt
for submission or evaluation purposes.git clone https://github.com/AI-FDU/ai-fdu.github.io.git
cd ai-fdu.github.io/pj3
mkdir model
To start the training process, execute:
python train.py
Ensure you have configured the dataset paths and any other necessary parameters within train.py
according to your setup.
Once training is complete, you can test the model on a designated test set by running:
python test.py > ans.txt
This command runs the testing script and redirects its output to ans.txt
, which typically contains the predicted counts per image or video frame.
Final Score: max(20, s1+s2+s3+s4), total of 20 points.
Competition Deadline: June 2, 2024 ‼️ Submit A-board’s best ranking and score screenshots, txt files to eLearning.
Report Deadline: June 14, 2024 ‼️ Submit B-board result screenshots, final code, project report (4 pages), PPT (if applicable) to eLearning.
Please ensure submission in accordance with the following: