diff --git a/configs/mot/README.md b/configs/mot/README.md index bf06d24f33f3041e9b5b1fb66bc0ae0050fbcec1..b543e5a72d0416530d7274587aafed1ba50848e2 100644 --- a/configs/mot/README.md +++ b/configs/mot/README.md @@ -119,6 +119,8 @@ If you use a stronger detection model, you can get better results. Each txt is t | :--------------| :------- | :----: | :----: | :----: | :----: | :----: | :------: | :----: |:-----: | | DLA-34(paper) | 1088x608 | 83.3 | 81.9 | 544 | 3822 | 14095 | - | - | - | | DLA-34 | 1088x608 | 83.2 | 83.1 | 499 | 3861 | 14223 | - | [model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 80.8 | 81.1 | 561 | 3643 | 16967 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 74.0 | 76.1 | 640 | 4989 | 23034 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | ### FairMOT Results on MOT-16 Test Set @@ -127,9 +129,11 @@ If you use a stronger detection model, you can get better results. Each txt is t | :--------------| :------- | :----: | :----: | :----: | :----: | :----: | :------: | :----: |:-----: | | DLA-34(paper) | 1088x608 | 74.9 | 72.8 | 1074 | - | - | 25.9 | - | - | | DLA-34 | 1088x608 | 75.0 | 74.7 | 919 | 7934 | 36747 | - | [model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 73.0 | 72.6 | 977 | 7578 | 40601 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 69.9 | 70.2 | 1044 | 8869 | 44898 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | **Notes:** - FairMOT used 2 GPUs for training and mini-batch size as 6 on each GPU, and trained for 30 epoches. + FairMOT DLA-34 used 2 GPUs for training and mini-batch size as 6 on each GPU, and trained for 30 epoches. ## Feature Tracking Model diff --git a/configs/mot/README_cn.md b/configs/mot/README_cn.md index e78e8e09d84e602b92c66c3d618811976e287f88..86b3ea66e7b6e37c5d034a41eefaa8aa544f8b82 100644 --- a/configs/mot/README_cn.md +++ b/configs/mot/README_cn.md @@ -120,6 +120,8 @@ wget https://dataset.bj.bcebos.com/mot/det_results_dir.zip | :--------------| :------- | :----: | :----: | :---: | :----: | :---: | :------: | :----: |:----: | | DLA-34(paper) | 1088x608 | 83.3 | 81.9 | 544 | 3822 | 14095 | - | - | - | | DLA-34 | 1088x608 | 83.2 | 83.1 | 499 | 3861 | 14223 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 80.8 | 81.1 | 561 | 3643 | 16967 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 74.0 | 76.1 | 640 | 4989 | 23034 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | ### FairMOT在MOT-16 Test Set上结果 @@ -127,9 +129,11 @@ wget https://dataset.bj.bcebos.com/mot/det_results_dir.zip | :--------------| :------- | :----: | :----: | :----: | :----: | :----: |:-------: | :----: | :----: | | DLA-34(paper) | 1088x608 | 74.9 | 72.8 | 1074 | - | - | 25.9 | - | - | | DLA-34 | 1088x608 | 75.0 | 74.7 | 919 | 7934 | 36747 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 73.0 | 72.6 | 977 | 7578 | 40601 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 69.9 | 70.2 | 1044 | 8869 | 44898 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | **注意:** - FairMOT使用2个GPU进行训练，每个GPU上batch size为6，训练30个epoch。 + FairMOT DLA-34均使用2个GPU进行训练，每个GPU上batch size为6，训练30个epoch。 ## 特色垂类跟踪模型 diff --git a/configs/mot/fairmot/README.md b/configs/mot/fairmot/README.md index 719f91ef131723d21e7f6cd10a9cf7b14595516d..9b8934162497c80a72fe2573f6e23622c0b51158 100644 --- a/configs/mot/fairmot/README.md +++ b/configs/mot/fairmot/README.md @@ -20,6 +20,8 @@ English | [简体中文](README_cn.md) | :--------------| :------- | :----: | :----: | :----: | :----: | :----: | :------: | :----: |:-----: | | DLA-34(paper) | 1088x608 | 83.3 | 81.9 | 544 | 3822 | 14095 | - | - | - | | DLA-34 | 1088x608 | 83.2 | 83.1 | 499 | 3861 | 14223 | - | [model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 80.8 | 81.1 | 561 | 3643 | 16967 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 74.0 | 76.1 | 640 | 4989 | 23034 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | ### FairMOT Results on MOT-16 Test Set @@ -28,9 +30,11 @@ English | [简体中文](README_cn.md) | :--------------| :------- | :----: | :----: | :----: | :----: | :----: | :------: | :----: |:-----: | | DLA-34(paper) | 1088x608 | 74.9 | 72.8 | 1074 | - | - | 25.9 | - | - | | DLA-34 | 1088x608 | 75.0 | 74.7 | 919 | 7934 | 36747 | - | [model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 73.0 | 72.6 | 977 | 7578 | 40601 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 69.9 | 70.2 | 1044 | 8869 | 44898 | - |[model](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | **Notes:** - FairMOT used 2 GPUs for training and mini-batch size as 6 on each GPU, and trained for 30 epoches. + FairMOT DLA-34 used 2 GPUs for training and mini-batch size as 6 on each GPU, and trained for 30 epoches. ## Getting Start diff --git a/configs/mot/fairmot/README_cn.md b/configs/mot/fairmot/README_cn.md index f4c958e5fc7d0390690ddbb595ae811cf819cdd8..a4675ddea92625d34ec396b7d8cf905d6721c885 100644 --- a/configs/mot/fairmot/README_cn.md +++ b/configs/mot/fairmot/README_cn.md @@ -20,6 +20,8 @@ | :--------------| :------- | :----: | :----: | :---: | :----: | :---: | :------: | :----: |:----: | | DLA-34(paper) | 1088x608 | 83.3 | 81.9 | 544 | 3822 | 14095 | - | - | - | | DLA-34 | 1088x608 | 83.2 | 83.1 | 499 | 3861 | 14223 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 80.8 | 81.1 | 561 | 3643 | 16967 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 74.0 | 76.1 | 640 | 4989 | 23034 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | ### FairMOT在MOT-16 Test Set上结果 @@ -27,9 +29,11 @@ | :--------------| :------- | :----: | :----: | :----: | :----: | :----: |:-------: | :----: | :----: | | DLA-34(paper) | 1088x608 | 74.9 | 72.8 | 1074 | - | - | 25.9 | - | - | | DLA-34 | 1088x608 | 75.0 | 74.7 | 919 | 7934 | 36747 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_1088x608.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml) | +| DLA-34 | 864x480 | 73.0 | 72.6 | 977 | 7578 | 40601 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_864x480.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml) | +| DLA-34 | 576x320 | 69.9 | 70.2 | 1044 | 8869 | 44898 | - |[下载链接](https://paddledet.bj.bcebos.com/models/mot/fairmot_dla34_30e_576x320.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml) | **注意:** - FairMOT使用2个GPU进行训练，每个GPU上batch size为6，训练30个epoch。 + FairMOT DLA-34均使用2个GPU进行训练，每个GPU上batch size为6，训练30个epoch。 ## 快速开始 diff --git a/configs/mot/fairmot/_base_/fairmot_reader_576x320.yml b/configs/mot/fairmot/_base_/fairmot_reader_576x320.yml new file mode 100644 index 0000000000000000000000000000000000000000..5198e3f33534f9b7708276f077468ed6a253fe0c --- /dev/null +++ b/configs/mot/fairmot/_base_/fairmot_reader_576x320.yml @@ -0,0 +1,41 @@ +worker_num: 4 +TrainReader: + inputs_def: + image_shape: [3, 320, 576] + sample_transforms: + - Decode: {} + - RGBReverse: {} + - AugmentHSV: {} + - LetterBoxResize: {target_size: [320, 576]} + - MOTRandomAffine: {reject_outside: False} + - RandomFlip: {} + - BboxXYXY2XYWH: {} + - NormalizeBox: {} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - RGBReverse: {} + - Permute: {} + batch_transforms: + - Gt2FairMOTTarget: {} + batch_size: 6 + shuffle: True + drop_last: True + use_shared_memory: True + +EvalMOTReader: + sample_transforms: + - Decode: {} + - LetterBoxResize: {target_size: [320, 576]} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - Permute: {} + batch_size: 1 + + +TestMOTReader: + inputs_def: + image_shape: [3, 320, 576] + sample_transforms: + - Decode: {} + - LetterBoxResize: {target_size: [320, 576]} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - Permute: {} + batch_size: 1 diff --git a/configs/mot/fairmot/_base_/fairmot_reader_864x480.yml b/configs/mot/fairmot/_base_/fairmot_reader_864x480.yml new file mode 100644 index 0000000000000000000000000000000000000000..33438a8d56b7bb911198838cb6878ee3a8b9d8a4 --- /dev/null +++ b/configs/mot/fairmot/_base_/fairmot_reader_864x480.yml @@ -0,0 +1,41 @@ +worker_num: 4 +TrainReader: + inputs_def: + image_shape: [3, 480, 864] + sample_transforms: + - Decode: {} + - RGBReverse: {} + - AugmentHSV: {} + - LetterBoxResize: {target_size: [480, 864]} + - MOTRandomAffine: {reject_outside: False} + - RandomFlip: {} + - BboxXYXY2XYWH: {} + - NormalizeBox: {} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - RGBReverse: {} + - Permute: {} + batch_transforms: + - Gt2FairMOTTarget: {} + batch_size: 6 + shuffle: True + drop_last: True + use_shared_memory: True + +EvalMOTReader: + sample_transforms: + - Decode: {} + - LetterBoxResize: {target_size: [480, 864]} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - Permute: {} + batch_size: 1 + + +TestMOTReader: + inputs_def: + image_shape: [3, 480, 864] + sample_transforms: + - Decode: {} + - LetterBoxResize: {target_size: [480, 864]} + - NormalizeImage: {mean: [0, 0, 0], std: [1, 1, 1]} + - Permute: {} + batch_size: 1 diff --git a/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml b/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml index 8c1e708c01a9b0b0b32d16d9605095e04a5ff176..3ef2b55c4e15ac8f3e4a484947c3d11fb8fb9d02 100644 --- a/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml +++ b/configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml @@ -6,5 +6,4 @@ _BASE_: [ '_base_/fairmot_reader_1088x608.yml', ] -metric: MOT weights: output/fairmot_dla34_30e_1088x608/model_final diff --git a/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml b/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml new file mode 100644 index 0000000000000000000000000000000000000000..e2f9ca5fde20c3601b5c4eb009e26a3359f6b5e5 --- /dev/null +++ b/configs/mot/fairmot/fairmot_dla34_30e_576x320.yml @@ -0,0 +1,9 @@ +_BASE_: [ + '../../datasets/mot.yml', + '../../runtime.yml', + '_base_/optimizer_30e.yml', + '_base_/fairmot_dla34.yml', + '_base_/fairmot_reader_576x320.yml', +] + +weights: output/fairmot_dla34_30e_576x320/model_final diff --git a/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml b/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml new file mode 100644 index 0000000000000000000000000000000000000000..8bc152d040a11b0e809b63a3ad0f4a3e2da492b6 --- /dev/null +++ b/configs/mot/fairmot/fairmot_dla34_30e_864x480.yml @@ -0,0 +1,9 @@ +_BASE_: [ + '../../datasets/mot.yml', + '../../runtime.yml', + '_base_/optimizer_30e.yml', + '_base_/fairmot_dla34.yml', + '_base_/fairmot_reader_864x480.yml', +] + +weights: output/fairmot_dla34_30e_864x480/model_final