Skip to content

Commit 7a0a984

Browse files
JackWoo0831JackWoo0831
authored
update codes and files of UAVMOT
1 parent 979d92f commit 7a0a984

File tree

2 files changed

+40
-3
lines changed

2 files changed

+40
-3
lines changed

‎run_yolov7.txt

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,18 @@ python tools/convert_VisDrone_to_yolov2.py --split_name VisDrone2019-MOT-test-de
1515
## car only and half dataset
1616
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
1717
## all category
18-
python train_aux.py --dataset visdrone --workers 8 --device 4 --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
18+
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
19+
20+
21+
## all category, no mosaic, rect training
22+
NOTE: edited augment = False in line 246, train_aux.py
23+
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
24+
25+
## all category, high resolution
26+
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
27+
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
28+
29+
1930

2031
launch tensorboard: tensorboard --logdir=$./runs/train/yolov7-w6-custom2$ --port=8090
2132

@@ -36,4 +47,7 @@ python tracker/track.py --dataset visdrone --data_format origin --tracker bytetr
3647
python tracker/track.py --dataset visdrone --data_format origin --tracker deepmot --model_path runs/train/yolov7-w6-custom4/weights/best.pt
3748

3849
## bot sort
39-
python tracker/track.py --dataset visdrone --data_format origin --tracker botsort --model_path runs/train/yolov7-w6-custom4/weights/best.pt
50+
python tracker/track.py --dataset visdrone --data_format origin --tracker botsort --model_path runs/train/yolov7-w6-custom4/weights/best.pt
51+
52+
## uavmot
53+
python tracker/track.py --dataset visdrone --data_format origin --tracker uavmot --model_path runs/train/yolov7-w6-custom4/weights/best.pt

‎track_result.txt

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,4 +111,27 @@ uav0000297_02761_v.txt 50.3% 69.8% 39.3% 42.8% 76.0% 168 5 97 66 3960 1
111111
uav0000306_00230_v.txt 46.2% 79.6% 32.6% 33.7% 82.3% 124 11 48 65 1061 9726 12 1535 26.4% 0.258 16 9 14
112112
uav0000355_00001_v.txt 46.0% 80.3% 32.3% 32.9% 81.7% 76 6 43 27 1046 9559 16 1610 25.4% 0.227 9 5 0
113113
uav0000370_00001_v.txt 36.7% 70.8% 24.8% 27.8% 79.4% 25 0 17 8 179 1787 6 363 20.3% 0.262 0 6 0
114-
OVERALL 41.4% 71.2% 29.1% 32.1% 78.4% 2293 155 1027 1111 27156 208980 1014 34440 23.0% 0.265 580 318 222
114+
OVERALL 41.4% 71.2% 29.1% 32.1% 78.4% 2293 155 1027 1111 27156 208980 1014 34440 23.0% 0.265 580 318 222
115+
116+
117+
UAVMOT
118+
average fps: 18.56506254197595
119+
IDF1 IDP IDR Rcll Prcn GT MT PT ML FP FN IDs FM MOTA MOTP IDt IDa IDm
120+
uav0000009_03358_v.txt 45.0% 53.2% 39.0% 57.3% 78.3% 118 43 45 30 1996 5347 262 685 39.3% 0.275 139 129 17
121+
uav0000073_00600_v.txt 30.9% 76.7% 19.4% 21.5% 84.9% 103 6 28 69 544 11242 25 239 17.5% 0.354 19 11 9
122+
uav0000073_04464_v.txt 23.2% 59.2% 14.4% 17.5% 71.9% 161 11 51 99 2356 28318 72 541 10.5% 0.354 46 26 16
123+
uav0000077_00720_v.txt 43.5% 64.6% 32.9% 46.3% 90.9% 90 22 40 28 892 10342 67 444 41.3% 0.218 20 56 11
124+
uav0000088_00290_v.txt 33.1% 41.6% 27.4% 46.3% 70.4% 119 15 69 35 4274 11766 359 1292 25.2% 0.332 229 129 23
125+
uav0000119_02301_v.txt 50.2% 95.0% 34.1% 34.8% 96.8% 51 11 11 29 67 3856 1 88 33.7% 0.220 1 1 1
126+
uav0000120_04775_v.txt 39.7% 61.4% 29.3% 35.6% 74.8% 329 40 95 194 5120 27455 143 1104 23.3% 0.249 67 97 27
127+
uav0000161_00000_v.txt 36.1% 76.6% 23.6% 25.0% 81.0% 193 23 38 132 1856 23784 39 568 19.0% 0.260 31 18 14
128+
uav0000188_00000_v.txt 29.0% 83.6% 17.5% 18.4% 87.5% 132 9 28 95 502 15681 6 251 15.7% 0.318 8 6 8
129+
uav0000201_00000_v.txt 36.0% 51.6% 27.6% 37.8% 70.5% 88 6 42 40 3231 12746 111 822 21.4% 0.299 19 88 5
130+
uav0000249_00001_v.txt 52.6% 55.8% 49.9% 63.5% 71.0% 173 65 47 61 2203 3103 105 596 36.4% 0.262 39 81 16
131+
uav0000249_02688_v.txt 46.0% 53.3% 40.4% 55.7% 73.5% 168 46 64 58 1467 3244 130 481 33.9% 0.264 78 72 22
132+
uav0000297_00000_v.txt 48.9% 57.8% 42.4% 50.5% 68.7% 175 50 52 73 2043 4409 133 494 26.0% 0.251 91 54 23
133+
uav0000297_02761_v.txt 55.4% 71.2% 45.3% 49.0% 77.1% 168 39 65 64 4269 14967 137 1266 34.0% 0.269 74 49 17
134+
uav0000306_00230_v.txt 50.5% 77.8% 37.3% 38.7% 80.7% 124 23 37 64 1355 8985 29 382 29.3% 0.281 14 21 9
135+
uav0000355_00001_v.txt 51.4% 80.3% 37.8% 38.9% 82.7% 76 30 19 27 1157 8695 17 388 30.7% 0.249 10 8 1
136+
uav0000370_00001_v.txt 39.4% 61.6% 29.0% 36.1% 76.7% 25 4 14 7 271 1582 8 61 24.8% 0.272 0 8 0
137+
OVERALL 40.5% 62.9% 29.8% 36.5% 77.0% 2293 443 745 1105 33603 195522 1644 9702 25.0% 0.276 885 854 219

0 commit comments

Comments
 (0)