# Mahdi Abdollahpour, 30/12/2021, 07:47 PM, PyCharm, ByteTrack
from yolox.core import launch
from yolox.data import MOTDataset, ValTransform, ValTransformWithPseudo
test_size = (896, 1600)
import os
from yolox.data import get_yolox_datadir
if __name__ == "__main__":
    valdataset = MOTDataset(
        data_dir=os.path.join(get_yolox_datadir(), "MOT20"),
        json_file='train_MOT20-01.json',
        img_size=test_size,
        name='train',  # change to train when running on training set
        preproc=ValTransformWithPseudo(
            rgb_means=(0.485, 0.456, 0.406),
            std=(0.229, 0.224, 0.225),
        ),
        load_weak=True
    )


    for batch in valdataset:
        print(batch)
        exit()