ipl-uw / 2019-CVPR-AIC-Track-2-UWIPL

Repository for 2019 CVPR AI City Challenge Track 2 from IPL@UW
MIT License
21 stars 12 forks source link
aicitychallenge computer-vision cvpr2019 reidentification vehicle-reid

2019-CVPR-AIC-Track-2-UWIPL

Repository for 2019 CVPR AI City Challenge Track 2 from IPL @University of Washington. Our method ranks 2nd in the competition.

Code structure

Our code consists of the following three components:

1. Video-Person-ReID

The multi-view and metadata re-ranking vehicle reidentification model. The code is based on Jiyang Gao's Video-Person-ReID [code].

2. Metadata

Metadata model for vehicle's type, brand and color. The code is based on [code].

3. CarKeypoints

The vehicle keypoints code is based on krrish94's CarKeypoints [code].

Training

Training of both Video-Person-ReID and metadata requires CarKeypoints's inference result on training set. For CarKeypoints, we use the pre-trained model [model]. Please refer to the README.md files in each subfolder.

Testing

Testing of both Video-Person-ReID and metadata requires CarKeypoints's inference result on testing set. In addition, Video-Person-ReID needs metadata's inference result on testing set.