Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Docs] add reid training README.md #210

Merged
merged 5 commits into from
Jul 14, 2021
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix some bug
  • Loading branch information
ToumaKazusa3 committed Jul 14, 2021
commit ab606e29405bdb42e1f28b53fb9a732fac498acf
2 changes: 1 addition & 1 deletion configs/mot/deepsort/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

## Results and models on MOT17

We implement SORT and DeepSORT with independent detector and ReID models. To train a model by yourself, you need to train a detector following [here](../../det/) and also train a ReID model.
We implement SORT and DeepSORT with independent detector and ReID models. To train a model by yourself, you need to train a detector following [here](../../det/) and also train a ReID model following [here](../../reid).
The configs in this folder are basically for inference.

Currently we do not support training ReID models.
Expand Down
13 changes: 10 additions & 3 deletions configs/reid/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

You may want to train a ReID model for multiple object tracking or other applications. We support ReID model training in MMTracking, which is built upon [MMClassification](https://github.com/open-mmlab/mmclassification).

## Standard Dataset
## 1.Standard Dataset

This section will show how to train a ReID model on standard datasets i.e. MOT17.

Expand All @@ -14,6 +14,11 @@ We need to download datasets following [here](https://github.com/open-mmlab/mmtr
python ./tools/convert_datasets/mot2reid.py -i ./data/MOT17/ -o ./data/MOT17/reid --val-split 0.2 --vis-threshold 0.3
```

Arguments:

- `--val-split`: Proportion of the validation dataset to the whole ReID dataset.
- `--vis-threshold`: Threshold of visibility for each person.

The directory of the converted datasets is as follows:

```
Expand All @@ -35,6 +40,8 @@ MOT17
│ │ ├── val_20.txt
```

Note: `80` in `train_80.txt` means the proportion of the training dataset to the whole ReID dataset is eighty percent. While the proportion of the validation dataset is twenty percent.

For training, we provide a annotation list `train_80.txt`. Each line of the list contrains a filename and its corresponding ground-truth labels. The format is as follows:

```
Expand Down Expand Up @@ -74,7 +81,7 @@ Optional arguments remain the same as stated above.

For more training details, please refer to [here](https://github.com/open-mmlab/mmtracking/blob/master/docs/quick_run.md).

### Customize Dataset
## 2.Customize Dataset

This section will show how to train a ReID model on customize datasets.

Expand Down Expand Up @@ -126,7 +133,7 @@ data = dict(
data_prefix='data/Filelist/imgs',
ann_file='data/Filelist/meta/val.txt'),
)
# modify number of classes, assume your training set has 100 classes
# modify the number of classes, assume your training set has 100 classes
model = dict(reid=dict(head=dict(num_classes=100)))
```

Expand Down