SOTDrawRect
The purpose of this repo is to provide evaluation API of Current Single Object Tracking Dataset, including
- VOT2016
- VOT2018
- VOT2018-LT
- OTB100(OTB2015)
- UAV123
- NFS
- LaSOT
- TrackingNet (Evaluation on Server)
- GOT-10k (Evaluation on Server)
Install
git clone https://github.com/Giveupfree/SOTDrawRect.git
pip install -r requirements.txt
cd toolkit/utils/
python setup.py build_ext --inplace
Update toolkit(optional)
The contents of the entire toolkit folder can be replaced directly from pysot-toolkit
Download Dataset
Download json files used in our toolkit baidu pan or Google Drive
-
Put CVRP13.json, OTB100.json, OTB50.json in OTB100 dataset directory (you need to copy Jogging to Jogging-1 and Jogging-2, and copy Skating2 to Skating2-1 and Skating2-2 or using softlink)
The directory should have the below format
| -- OTB100/
| -- Basketball
| ......
| -- Woman
| -- OTB100.json
| -- OTB50.json
| -- CVPR13.json
-
Put all other jsons in the dataset directory like in step 1
Usage
Draw rectangular boxes
cd /path/to/SOTDrawRect
python draw_rect.py \
--dataset_dir /path/to/dataset/root \ # dataset path
--dataset VOT2018 \ # dataset name(VOT2018, VOT2016, OTB100, GOT10k)
--tracker_result_dir /path/to/tracker/dir \ # tracker dir
--format pdf \ # save fomat (pdf,png,jpg)
--trackers ours ECO UPDT SiamRPNpp \ # tracker names
--save_dir \ # save dir
Draw a bounding box for a video sequence
cd /path/to/SOTDrawRect
python draw_rect.py \
-- video videoname \
--dataset_dir /path/to/dataset/root \ # dataset path
--dataset VOT2018 \ # dataset name(VOT2018, VOT2016, OTB100, GOT10k)
--tracker_result_dir /path/to/tracker/dir \ # tracker dir
--format pdf \ # save fomat (pdf,png,jpg)
--trackers ours ECO UPDT SiamRPNpp \ # tracker names
--save_dir \ # save dir