Skip to content

Commit 11aeb8b

Browse files
karineipkarineip
authored
Kaipkiun/02 bug fixes (#594)
1 parent 777856b commit 11aeb8b

File tree

2 files changed

+27
-22
lines changed

2 files changed

+27
-22
lines changed

‎scenarios/tracking/02_mot_challenge.ipynb‎

100644100755
Lines changed: 22 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@
103103
},
104104
{
105105
"cell_type": "code",
106-
"execution_count": 3,
106+
"execution_count": 9,
107107
"metadata": {
108108
"tags": [
109109
"parameters"
@@ -121,7 +121,6 @@
121121
"source": [
122122
"CONF_THRES = 0.4\n",
123123
"TRACK_BUFFER = 30\n",
124-
"IM_SIZE = (1080, 1920)\n",
125124
"\n",
126125
"# Downloaded MOT Challendage data path\n",
127126
"MOT_ROOT_PATH = \"../../data/\"\n",
@@ -243,37 +242,43 @@
243242
"name": "stdout",
244243
"output_type": "stream",
245244
"text": [
245+
"loaded ./models/all_dla34.pth, epoch 10\n",
246246
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-02-SDP.txt\n",
247247
"Evaluate seq: MOT17-02-SDP\n",
248+
"loaded ./models/all_dla34.pth, epoch 10\n",
248249
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-04-SDP.txt\n",
249250
"Evaluate seq: MOT17-04-SDP\n",
251+
"loaded ./models/all_dla34.pth, epoch 10\n",
250252
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-05-SDP.txt\n",
251253
"Evaluate seq: MOT17-05-SDP\n",
254+
"loaded ./models/all_dla34.pth, epoch 10\n",
252255
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-09-SDP.txt\n",
253256
"Evaluate seq: MOT17-09-SDP\n",
257+
"loaded ./models/all_dla34.pth, epoch 10\n",
254258
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-10-SDP.txt\n",
255259
"Evaluate seq: MOT17-10-SDP\n",
260+
"loaded ./models/all_dla34.pth, epoch 10\n",
256261
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-11-SDP.txt\n",
257262
"Evaluate seq: MOT17-11-SDP\n",
263+
"loaded ./models/all_dla34.pth, epoch 10\n",
258264
"Saved tracking results to ./results/MOT_val_all_dla34/MOT17-13-SDP.txt\n",
259265
"Evaluate seq: MOT17-13-SDP\n",
260-
" IDF1 IDP IDR Rcll Prcn GT MT PT ML FP FN IDs FM MOTA MOTP IDt IDa IDm\n",
261-
"MOT17-02-SDP 60.1% 68.9% 53.3% 72.3% 93.4% 62 28 25 9 950 5148 243 628 65.9% 0.198 138 49 16\n",
262-
"MOT17-04-SDP 83.2% 85.1% 81.2% 86.3% 90.4% 83 51 21 11 4348 6526 32 219 77.1% 0.172 6 20 1\n",
263-
"MOT17-05-SDP 73.1% 75.5% 70.9% 83.9% 89.2% 133 74 50 9 702 1117 120 217 72.0% 0.221 100 45 41\n",
264-
"MOT17-09-SDP 65.9% 71.6% 61.0% 81.5% 95.7% 26 18 8 0 197 984 45 105 77.0% 0.175 34 10 7\n",
265-
"MOT17-10-SDP 63.5% 68.4% 59.2% 79.0% 91.3% 57 30 27 0 970 2700 182 439 70.0% 0.233 109 44 14\n",
266-
"MOT17-11-SDP 82.2% 82.0% 82.4% 92.3% 91.9% 75 55 16 4 764 722 75 112 83.5% 0.171 29 34 9\n",
267-
"MOT17-13-SDP 64.6% 65.9% 63.3% 77.2% 80.3% 110 50 51 9 2203 2658 164 457 56.8% 0.279 97 39 42\n",
268-
"OVERALL 73.9% 77.5% 70.7% 82.3% 90.1% 546 306 198 42 10134 19855 861 2177 72.5% 0.196 513 241 130\n"
266+
" IDF1 IDP IDR Rcll Prcn GT MT PT ML FP FN IDs FM MOTA MOTP IDt IDa IDm\n",
267+
"MOT17-02-SDP 63.9% 77.4% 54.4% 68.5% 97.4% 62 22 31 9 344 5855 183 656 65.7% 0.193 98 33 13\n",
268+
"MOT17-04-SDP 83.7% 86.1% 81.3% 86.3% 91.4% 83 51 20 12 3868 6531 32 201 78.1% 0.171 5 20 2\n",
269+
"MOT17-05-SDP 75.9% 82.9% 70.1% 81.2% 96.0% 133 63 59 11 236 1303 79 207 76.6% 0.199 83 26 40\n",
270+
"MOT17-09-SDP 65.4% 71.6% 60.2% 81.1% 96.5% 26 19 7 0 158 1006 52 105 77.2% 0.165 37 12 7\n",
271+
"MOT17-10-SDP 65.0% 72.1% 59.2% 78.8% 96.0% 57 32 25 0 418 2721 149 404 74.4% 0.213 89 43 14\n",
272+
"MOT17-11-SDP 85.6% 87.9% 83.4% 90.4% 95.2% 75 52 19 4 426 910 38 134 85.4% 0.157 24 19 13\n",
273+
"MOT17-13-SDP 77.0% 82.0% 72.5% 83.9% 94.8% 110 74 29 7 534 1878 86 373 78.5% 0.205 72 26 35\n",
274+
"OVERALL 76.8% 82.3% 71.9% 82.0% 93.9% 546 313 190 43 5984 20204 619 2080 76.1% 0.182 408 179 124\n"
269275
]
270276
}
271277
],
272278
"source": [
273279
"strsummary = tracker.eval_mot(\n",
274280
" conf_thres=CONF_THRES,\n",
275-
" track_buffer=TRACK_BUFFER,\n",
276-
" im_size=IM_SIZE,\n",
281+
" track_buffer=TRACK_BUFFER, \n",
277282
" data_root=mot_train_path,\n",
278283
" seqs=train_seqs,\n",
279284
" result_root=RESULT_ROOT,\n",
@@ -327,8 +332,7 @@
327332
"source": [
328333
"tracker.eval_mot(\n",
329334
" conf_thres=CONF_THRES,\n",
330-
" track_buffer=TRACK_BUFFER,\n",
331-
" im_size=IM_SIZE,\n",
335+
" track_buffer=TRACK_BUFFER, \n",
332336
" data_root=mot_test_path,\n",
333337
" seqs=test_seqs,\n",
334338
" result_root=RESULT_ROOT,\n",
@@ -354,7 +358,7 @@
354358
"name": "python",
355359
"nbconvert_exporter": "python",
356360
"pygments_lexer": "ipython3",
357-
"version": "3.8.1"
361+
"version": "3.7.6"
358362
},
359363
"toc": {
360364
"base_numbering": 1,
@@ -369,10 +373,10 @@
369373
"height": "calc(100% - 180px)",
370374
"left": "10px",
371375
"top": "150px",
372-
"width": "356.263px"
376+
"width": "356.258px"
373377
},
374378
"toc_section_display": true,
375-
"toc_window_display": true
379+
"toc_window_display": false
376380
}
377381
},
378382
"nbformat": 4,

‎utils_cv/tracking/model.py‎

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,6 @@ def eval_mot(
300300
self,
301301
conf_thres: float,
302302
track_buffer: int,
303-
im_size: Tuple[int, int],
304303
data_root: str,
305304
seqs: list,
306305
result_root: str,
@@ -312,7 +311,6 @@ def eval_mot(
312311
Args:
313312
conf_thres: confidence thresh for tracking
314313
track_buffer: tracking buffer
315-
im_size: image resolution
316314
data_root: data root path
317315
seqs: list of video sequences subfolder names under MOT challenge data
318316
result_root: tracking result path
@@ -326,7 +324,7 @@ def eval_mot(
326324
if not osp.exists(eval_path):
327325
os.makedirs(eval_path)
328326

329-
#Loop over all video sequences
327+
# Loop over all video sequences
330328
for seq in seqs:
331329
result_filename = "{}.txt".format(seq)
332330
im_path = osp.join(data_root, seq, "img1")
@@ -345,7 +343,10 @@ def eval_mot(
345343
# Run model inference
346344
if not osp.exists(result_path):
347345
eval_results = self.predict(
348-
im_path, conf_thres, track_buffer, im_size, frame_rate
346+
im_or_video_path=im_path,
347+
conf_thres=conf_thres,
348+
track_buffer=track_buffer,
349+
frame_rate=frame_rate,
349350
)
350351
result_path = savetxt_results(
351352
eval_results, exp_name, result_root, result_filename

0 commit comments

Comments
 (0)