Environment

FairMOT

• pytorch 1.2

Transformer

• pytorch 1.8

Usage

git clone https://github.com/dlutor/cranesafety.git

FairMOT

Download dataset from Datasets.

Move and extract to ./FairMOT/datasets.

Download pretrained model from pretrained.

Move and extract to ./FairMOT/models.

Train

cd FairMOT
python ./src/train.py

Test

python demo.py

Transformer

Train and test

cd Trajectory-Transformer
python train_individualTF.py

or python train_newTF.py