forked from JackWoo0831/Yolov7-tracker
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_yolov7.txt
53 lines (37 loc) · 3.48 KB
/
run_yolov7.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
conda activate openmmlab
0.generate dataset
## car only and half dataset
python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-train --generate_imgs --car_only --half
python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-val --generate_imgs --car_only
python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-test-dev --generate_imgs --car_only
## all category
python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-train --generate_imgs
python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-val --generate_imgs
python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-test-dev --generate_imgs
1.train:
## car only and half dataset
python train_aux.py --dataset visdrone --workers 8 --device 4 --batch-size 16 --data data/visdrone_half_car.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights './weights/yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml
## all category
python train_aux.py --dataset visdrone --workers 8 --device 0 --batch-size 16 --data data/visdrone_all.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights './weights/yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml
## all category, no mosaic, rect training
NOTE: edited augment = False in line 246, train_aux.py
python train_aux.py --dataset visdrone --workers 8 --device 2 --batch-size 16 --data data/visdrone_all.yaml --img 1280 1280 --rect --cfg cfg/training/yolov7-w6.yaml --weights './weights/yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml
## all category, high resolution
python -m torch.distributed.launch --nproc_per_node 2 train_aux.py --dataset visdrone --workers 8 --device 0,1 --sync-bn --batch-size 16 --data data/visdrone_all.yaml --img 1920 1920 --cfg cfg/training/yolov7-w6.yaml --weights './weights/yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml
python train_aux.py --dataset visdrone --workers 8 --device 0 --batch-size 8 --data data/visdrone_all.yaml --img 1920 1920 --cfg cfg/training/yolov7-w6.yaml --weights './weights/yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml
launch tensorboard: tensorboard --logdir=$./runs/train/yolov7-w6-custom2$ --port=8090
2.test:
python test.py --dataset visdrone --data data/visdrone_half_car.yaml --img 1280 --batch 32 --conf 0.001 --iou 0.65 --task test --device 4 --save-txt --weights ./runs/train/yolov7-w6-custom3/weights/last.pt --name visdrone_yolov7w6_val
3.track:
## sort
python tracker/track.py --dataset visdrone --data_format origin --tracker sort --model_path runs/train/yolov7-w6-custom4/weights/best.pt
## deep sort
python tracker/track.py --dataset visdrone --data_format origin --tracker deepsort --model_path runs/train/yolov7-w6-custom4/weights/best.pt --save_images
## byte track
python tracker/track.py --dataset visdrone --data_format origin --tracker bytetrack --model_path runs/train/yolov7-w6-custom4/weights/best.pt --save_images --save_videos
## deep mot
python tracker/track.py --dataset visdrone --data_format origin --tracker deepmot --model_path runs/train/yolov7-w6-custom4/weights/best.pt
## bot sort
python tracker/track.py --dataset visdrone --data_format origin --tracker botsort --model_path runs/train/yolov7-w6-custom4/weights/best.pt
## uavmot
python tracker/track.py --dataset visdrone --data_format origin --tracker uavmot --model_path runs/train/yolov7-w6-custom4/weights/best.pt