Skip to content

Latest commit

 

History

History
35 lines (30 loc) · 660 Bytes

readme.md

File metadata and controls

35 lines (30 loc) · 660 Bytes
Raw

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