Towards Labeling-free Fine-grained Animal Pose Estimation(ACM MM'24)

This project is the implementation of our proposed method FreeNet paper. Our supplementary materials are available at: supplementary.

Abstract

In this paper, we are interested in identifying denser and finer animals joints. The lack of standardized joint definitions across various APE datasets, e.g., AnimalPose with 20 joints, AP-10k with 17 joints, and TigDog with 19 joints, presents a significant challenge yet offers an opportunity to fully utilize annotation data. This paper challenges this new non-standardized annotation problem, aiming to learn fine-grained (e.g., 24 or more joints) pose estimators in datasets that lack complete annotations. To combat the unannotated joints, we propose FreeNet, comprising a base network and an adaptation network connected through a circuit feedback learning paradigm. FreeNet enhances the adaptation network's tolerance to unannotated joints via body part-aware learning, optimizing the sampling frequency of joints based on joint detection difficulty, and improves the base network's predictions for unannotated joints using feedback learning. This leverages the cognitive differences of the adaptation network between non-standardized labeled and large-scale unlabeled data. Experimental results on three non-standard datasets demonstrate the effectiveness of our method for fine-grained APE.

Idea

We observe shared joints in different animal body parts exhibit different learning difficulties due to dataset gaps and considerable species variation. This accuracy difference between head joints and back body joints is close to 10%. Moreover, due to lacking full annotated data, the unshared joints across different datasets have the smallest average number and the worst accuracy, aggravating the skewed joint distribution in the combined training data.

To combat the unannotated joints, we chose bi-model training to leverage the different abilities of the two models in complementary ways. Specifically, we propose FreeNet, which consists of a base network and an adaptation network, aiming at fine-grained APE with free full annotation labels.

Datasets

AP-10K

AP-10K dataset is provided here.

AnimalPose Dataset

AnimalPose dataset could be downloaded through the official website.We randomly split the dataset by train:val = 9:1 for every category.

TigDog dataset

TigDog dataset could be downloaded here

Dataset Preparation

The coco-formated annotations of AnimalPose, TigDog and combined dataset are at freenet/resources/annotation_files. ap_10k_train/val/test.json are simply renamed from official files.

Then arrange your dataset directory as follows:

- Dataset
	- ap_10k
            - annotations
            - data
                - 1.jpg
                - ...
	- animal_pose
            - annotations
            - data
                - 1.jpg
                - ...
	- tigdog
            - annotations
            - data
                - horse
                    - 1.jpg
                    - ...
                - tiger
                    - 1.jpg
                    - ...
	- ap10k_animalpose
            - annotations
            - data
                - 1.jpg
                - ...
	- ap10k_animalpose_tigdog
            - annotations
            - data
                - 1.jpg
                - ...
                - horse
                    - 1.jpg
                    - ...
                - tiger
                    - 1.jpg
                    - ...
- freenet

Demo

Samples Using Different Loss

Install Conda Env

conda create -n freenet python=3.8
conda activate freenet
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
pip install -r requirments.txt

Result Test on AP-10K

Train on Synthetic Datasets Derived from AP-10k

Model	Shared Joint Size	mAP	PCK	Pretrained Weights	Download
Ours	5	72.70	80.57	weights	model
Ours	9	74.20	82.50	weights	model

Train on Three Synthesized Non-standard Datasets

Tips: These datasets are modified from “25 images per species” setting of ScarceNet.

Model	mAP	PCK	Pretrained Weights	Download
Ours	57.9	68.31	weights	model

Train on combined AP-10k and AnimalPose

Model	Random Sample Ratio	mAP	PCK	Pretrained Weights	Download
Ours	5%	48.6	65.3	weights	model
Ours	10%	57.5	71.28	weights	model
Ours	20%	63.4	75.4	weights	model

Tips:

Some other model weights like random-10%-ap10k used to draw exmaples in the first column of Fig6(a) are also provided here.

Inference Demo

In inference_example.py/inference_example.ipynb(jupyter), just run all or you can change the image or model file .

img_path = "test.jpg"
# Ls / + Lu / +Lf
weights_path = "saved_weights/ls_lu_lf.pth" #change it to the model you want to use

Train

For synthetic datasets derived from AP-10k:

# Parameter level is required here.
# Level = 0 -> share eyes,nose,neck,tail(5 kps)
# Level = 1 -> share Level 0 + paws(9 kps)
python ours_split_ap_10k.py --level 0 \
    --workers 8 \
    --seed 2 \
    --batch-size 32 \
    --mu 1 \
    --total-steps 45000 \
    --eval-step 150 \
    --warmup-steps 900 \
    --down-step 9000 \
    --feedback-steps-start 4500 \
    --feedback-steps-complete 6000 \
    --amp

For three synthesized non-standard datasets from 25->5 ips setting(AP-10k):

python ours_synthetic_ap_10k.py \
    --workers 8 \
    --seed 2 \
    --batch-size 8 \
    --mu 7 \
    --total-steps 45000 \
    --eval-step 150 \
    --warmup-steps 900 \
    --down-step 9000 \
    --feedback-steps-start 4500 \
    --feedback-steps-complete 6000 \
    --amp

python finetune.py \
    --workers 8 \
    --seed 2 \
    --batch-size 8 \
    --mu 7 \
    --total-steps 6000 \
    --eval-step 150 \
    --warmup-steps 900 \
    --down-step 6000 \
    --feedback-steps-start 4500 \
    --feedback-steps-complete 6000 \
    --amp

For 10% combined AP-10k and AnimalPose:

python ours_ap_10k_animal_pose.py \
    --workers 8 \
    --seed 2 \
    --batch-size 8 \
    --mu 7 \
    --total-steps 45000 \
    --eval-step 150 \
    --warmup-steps 900 \
    --down-step 9000 \
    --feedback-steps-start 4500 \
    --feedback-steps-complete 6000 \
    --amp

For 10% AP-10k + AnimalPose + TigDog:

python ours_ap10k_animalpose_tigdog.py \
    --workers 8 \
    --seed 2 \
    --batch-size 8 \
    --mu 7 \
    --total-steps 45000 \
    --eval-step 150 \
    --warmup-steps 900 \
    --down-step 9000 \
    --feedback-steps-start 4500 \
    --feedback-steps-complete 6000 \
    --amp

Test

For evaluation on AP-10k:

# remember to change the key order in codes if you directly use the checkpoint file
# for key in ['teacher_model','student_model','state_dict','model','directly']
python eval_ap10k.py \
    --cfg outer_tools/experiments/ap10k/hrnet/w32_256x192_adam_lr1e-3_ap10k.yaml \
    --animalpose \
    OUTPUT_DIR output/test \
    TEST.MODEL_FILE path_to_your_model \
    MODEL.NAME pose_hrnet \
    GPUS [0,]

For evaluation on AP-10k + AnimalPose

python eval_ap10k_animalpose.py \
    --cfg outer_tools/experiments/ap10k/hrnet/w32_256x192_adam_lr1e-3_ap10k_animalpose.yaml \
    --animalpose \
    OUTPUT_DIR output/test \
    TEST.MODEL_FILE path_to_your_model \
    MODEL.NAME pose_hrnet \
    GPUS [0,]

Acknowledgements

The code for network architecture, data preprocessing, and evaluation are adapted from ScarceNet and HRNet.

Contact

If you have any questions, please contact the authors:

Dan Zeng: zengd@sustech.edu.cn or danzeng1990@gmail.com

Yu Zhu: 12332483@mail.sustech.edu.cn

danzeng1990/FreeNet