Skip to content

Commit

Permalink
imrpove readme, remove unused dep, add some models
Browse files Browse the repository at this point in the history
  • Loading branch information
RaphaelLorenzo committed Dec 18, 2024
1 parent 67bff0c commit c6c9b93
Show file tree
Hide file tree
Showing 25 changed files with 25 additions and 3,195 deletions.
2 changes: 0 additions & 2 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,8 +1,6 @@
*.pyc
*.pth
*.pt
*.zip
*.deb
*.bin
*.onnx
output/
34 changes: 25 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@

Human tracking pipeline used in euROBIN system as described in XXX.

![Pipeline visu](images/eurobin_pipeline_gif.gif)

## Description

The pipeline consists of the following main components :
Expand All @@ -13,6 +15,17 @@ The pipeline consists of the following main components :
## Installation
How to install with docker and launch interactively

- Download models and place them in `./models/`
```
mkdir models
wget https://download.openmmlab.com/mmdetection/v2.0/yolo/yolov3_d53_320_273e_coco/yolov3_d53_320_273e_coco-421362b6.pth
mv yolov3_d53_320_273e_coco-421362b6.pth models/
```

Go to : [Onedrive](https://1drv.ms/u/s!AimBgYV7JjTlgccifT1XlGRatxg3vw?e=9wz7BY) and download the model then put in under `models/vitpose_small.pth`

The 6DRepNet models (face detection and 6D pose estimation) are already in `/models` (downloaded from [SixDRepNet/weights](https://github.com/jahongir7174/SixDRepNet/tree/master/weights))

- Build docker

```bash
Expand All @@ -25,6 +38,12 @@ docker build -t inria_docker:rgbd_detect .
sh launch.sh
```

- Launch the pipeline

```bash
python rgbd_detect_3d_dir.py -sixdrep
```

## ROS Interface
Input / output topics and format

Expand All @@ -47,16 +66,13 @@ See `python rgbd_detect_3d_dir.py -h`
Use `export VERBOSE=0` to set minimal verbosity (warnings, debug, errors and success). Level 1 adds info and 2 adds timing info.

## Acknowledgments
MMDet, MMPose, 6DRepNet...
The implementation is based on the following implementations of human detection and tracking, human pose estimation and 6D Head Pose estimation

- [ViTPose](https://github.com/ViTAE-Transformer/ViTPose)
- [MMDetection](https://github.com/open-mmlab/mmdetection)
- [MMPose](https://github.com/open-mmlab/mmpose)
- [6DRepNet](https://github.com/thohemp/6DRepNet) and [SixDRepNet](https://github.com/jahongir7174/SixDRepNet)

## TODO
- Clean the code of unnecesary options (keep only sixdrep)
- Remove unecessary dependencies
- Keep only rgbd_detect_3d_dir.py
- Add models in external links, LFS or directly from their source in the corresponding repos
- Add a description of all options
- Add ROS interface description
- Add illustration
- Add instructions and `requirements.txt` for local installation
- Add links and citations in Acknowledgments section
- Add instructions and `requirements.txt` for local installation
99 changes: 0 additions & 99 deletions lib/data/augmentation.py

This file was deleted.

136 changes: 0 additions & 136 deletions lib/data/datareader_h36m.py

This file was deleted.

59 changes: 0 additions & 59 deletions lib/data/datareader_mesh.py

This file was deleted.

Loading

0 comments on commit c6c9b93

Please sign in to comment.