HuggingFace DETR

HuggingFace DETR Training, Evaluation, Inferencing Guide

Clone repo

git clone [email protected]:ARG-NCTU/huggingface-detr.git

Enter the repo

cd ~/huggingface-detr/

Enter Docker Environment

For first terminal:

source Docker/ros1-gpu/run.sh

More terminal:

source Docker/ros1-gpu/join.sh

In addition, you can convert docker image to SQSH file (not necessary)

source docker2sqsh.sh 

Prepare Dataset

Download HuggingFace dataset:

huggingface-cli login
huggingface-cli download ARG-NCTU/Boat_dataset_2024 data --repo-type dataset --local-dir ~/huggingface-detr

Unzip images:

unzip ~/huggingface-detr/data/images.zip -d ~/huggingface-detr/

Training, Evaluation, Inferencing

Log in huggingface

huggingface-cli login

Enter the repo

cd ~/huggingface-detr/

Training

python3 train_detr_boat.py

Evaluation

python3 eval_detr_boat.py

Inferencing

python3 inference_detr_boat.py

Build ROS1 Workspace

Enter the repo

cd ~/huggingface-detr/

Setup ROS

source environment_ros1.sh 

Clean catkin ws

source clean_ros1_all.sh

Build catkin ws

source build_ros1_all.sh

ROS1 Inference

Setup ROS

source environment_ros1.sh 

Run DETR

roslaunch detr_inference detr_inference.launch