occlusion_eval

This is a project for a bachelor degree. Occlusion evaluation based on keypoints detection and modeling based on part relationship

The baseline of the project is AlphePose-pytorch

Environment

• Ubuntu 16.04/18.04.
• python 3.5+.
• Pytorch 1.0.0+.
• other dependencies.

Installation

1. Get the code.

git clone https://github.com/PanXF-HUST/occlusion_eval

2. Install pytorch 1.0.0and the corresponding version of torchvision and other dependencies.

pip install -r requirements.txt

3. Download the models manually: duc_se.pth (2018/08/30) (Google Drive | Baidu pan), yolov3-spp.weights(Google Drive | Baidu pan). Place them into ./models/sppe and ./models/yolo respectively.

Quick Start

• Input dir: Run Project for all images in a folder with:

python3 demo.py --indir ${img_directory} --outdir examples/res --save_img

NOTE

This project is a multi-task integration, part occlusion evaluation is only one part of the project, you can change the [fn.py] to change the format of output