security_vision/alphapose-feedback
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity
Pulling the traditional media art trick on training Alphapose.
5 commits 1 branch 0 tags 3.5 MiB
Jupyter Notebook 95.9%
Python 4%
Shell 0.1%
Find a file
Ruben van de Ven 96322c26e3 export diffs to a folder with videos and HTML index page
2023-04-21 17:25:37 +02:00
alphapose-docker@5aa2d0fac1 Alphapose-docker as submodule 2023-03-08 14:38:10 +01:00
.gitignore Alphapose-docker as submodule 2023-03-08 14:38:10 +01:00
.gitmodules Alphapose-docker as submodule 2023-03-08 14:38:10 +01:00
.python-version Scripts to run alphapose training in a loop and analyse results 2023-03-08 14:04:03 +01:00
downloadCOCO.sh Scripts to run alphapose training in a loop and analyse results 2023-03-08 14:04:03 +01:00
find_diffs.ipynb export diffs to a folder with videos and HTML index page 2023-04-21 17:25:37 +02:00
loop_alphapose_training.py Alphapose-docker as submodule 2023-03-08 14:38:10 +01:00
poetry.lock dependencies for loss calculation 2023-04-21 17:25:05 +02:00
pyproject.toml dependencies for loss calculation 2023-04-21 17:25:05 +02:00
README.md Mention COCO download 2023-03-08 14:48:29 +01:00
run_test_inferences.py Scripts to run alphapose training in a loop and analyse results 2023-03-08 14:04:03 +01:00

README.md

Looping Alphapose Training

Pulling the traditional media art trick: can we loop the training of Alphapose? What if we train it on COCO data, run it on COCO images, train it on these results, run it on COCO images etc. ad inf.?

Installation

poetry install
git submodule init
git submodule update

Then make sure alphapose-docker is set up:

cd alphapose-docker
docker build --tag alphapose .

As mentioned in the README from alphapose-docker it is necessary to download some external models before usage, as per the AlphaPose installation guide.

  1. YOLOv3 Object detector can be place in the alphapose-docker/detector/yolo/data repository.
  2. (Optionally) YOLOX models go in the alphapose-docker/detector/yolox/data repository.
  3. A pretrained AlphaPose model can be place in the alphapose-docker/pretrained_models directory. See their Model Zoo for the various options.
  4. For pose tracking, see the pose tracking module. Make sure to add the necessary folder as a volume to the docker run command.

Then make sure to download COCO keypoint 2017 to the data directory:

./downloadCOCO.sh

Usage

Then the scripts can be invoked with poetry run python ....py