code for IJCAI 18 paper: Grouping Attribute Recognition for Pedestrian with Joint Recurrent Learning
runtime environment
- python version : 3.4
- tensorflow version: >= 1.4
resource
best rap model is: 链接: https://pan.baidu.com/s/1m_bOW2x9_fyhy8dSlPWCuw 提取码: y9ip
best peta model is: 链接: https://pan.baidu.com/s/1acNTb668IVppJyhfzop_ow 提取码: 3upe
rap TFRecord file: 链接: https://pan.baidu.com/s/15MRSqdg7izo8HgnOD2uCTQ 提取码: 85uk
dataset label file: 链接: https://pan.baidu.com/s/16D1spNxN2SjZZR5l6v22Uw 提取码: iswu
prepare data (use rap as example)
- ROI data:
use pose estimation model provided in Spindle Net (github link is https://github.com/yokattame/SpindleNet) to get region proposal data.
- put rap or peta label data and region proposal data together.
file format is:
# 0
CAM12_2014-03-05_20140305110334-20140305111754_tarid1199_frame8675_line1.png
-1 -1 -1 1 -1 1 -1 1 -1 -1 -1 1 -1 -1 -1 1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 1 -1 -1 1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1
19.75390625 0 68 48.6875
0 28.796875 68 112.9609375
0 72.4375 68 188.8125
49.3738839286 29.6875 68 112.8125
0 31.3203125 31.921875 97.8203125
20.4285714286 85.796875 47.5714285714 186.5859375
8.18247767857 74.8125 35.3253348214 174.5625
that is
image index
attribute labels
head region coordinate
up region coordinate
down region coordinate
left arm coordinate
right arm coordinate
left leg coordinate
eight leg coordinate
- run data/build_rap_region_data.py to get tensorflow TFRecord input
train
- install bazel
- in root dir, run "bazel build //inception:rap_train" or "bazel build //inception:rap_test"
run command for example:
TRAIN_DIR=DIR for train models
RAP_DATA_DIR= DIR for rap tensorflow TFRecord file
MODEL_PATH=DIR for pretrained model
bazel-bin/inception/rap_train \
--train_dir="${TRAIN_DIR}" \
--data_dir="${RAP_DATA_DIR}" \
--pretrained_model_checkpoint_path="${MODEL_PATH}" \
--fine_tune=False \
--initial_learning_rate=0.1 \
--input_queue_memory_factor=1 \
--num_gpus=1 \
--max_steps=1001
test
TRAIN_DIR=DIR for train models
EVAL_DIR= DIR for eval event logs
RAP_DATA_DIR=DIR for rap tensorflow TFRecord file
bazel-bin/inception/rap_eval \
--eval_dir="${EVAL_DIR}" \
--data_dir="${RAP_DATA_DIR}" \
--subset=validation \
--num_examples=8317 \
--checkpoint_dir="${TRAIN_DIR}" \
--input_queue_memory_factor=1 \
--run_once