dnt 0.3.1.3__py3-none-any.whl → 0.3.1.7__py3-none-any.whl

dnt/track/tracker.py

@@ -1,16 +1,17 @@
-import os, sys
-sys.path.append(os.path.dirname(__file__))
-
-import torch
+import pandas as pd
+import os
+import sys
 from dsort import track as track_dsort
 from sort import track as track_sort
 from botsort import track as track_botsort
 from config import Config
-import pandas as pd
+
 from tqdm import tqdm
 import numpy as np
 from ..engine import interpolate_bbox, interpolate_bboxes, cluster_by_gap
 
+sys.path.append(os.path.dirname(__file__))
+
 class Tracker(object):
     def __init__(self, cfg: dict = None, device:str='auto', half:bool=False):
         '''
@@ -195,6 +196,9 @@ class Tracker(object):
     
         pbar.close()
         
+        if results == []:
+            return None
+        
         df = pd.concat(results, ignore_index=True)
         if inplace:
             tracks_del = tracks[~tracks['track'].isin(del_ids)].copy()

{dnt-0.3.1.3.dist-info → dnt-0.3.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: dnt
-Version: 0.3.1.3
+Version: 0.3.1.7
 Summary: A Python tool for video-based traffic analytiscs
 Home-page: https://its.cutr.usf.edu/dnt/
 Author: Zhenyu Wang
@@ -16,23 +16,22 @@ License-File: LICENSE
 Requires-Dist: torch
 Requires-Dist: torchvision
 Requires-Dist: torchaudio
-Requires-Dist: ultralytics >=8.2
+Requires-Dist: ultralytics>=8.2
 Requires-Dist: cython
-Requires-Dist: cython-bbox
+Requires-Dist: cython_bbox
 Requires-Dist: easydict
 Requires-Dist: faiss-cpu
-Requires-Dist: faiss-gpu
 Requires-Dist: filterpy
 Requires-Dist: gdown
 Requires-Dist: geopandas
 Requires-Dist: h5py
-Requires-Dist: lap
+Requires-Dist: lapx
 Requires-Dist: loguru
 Requires-Dist: matplotlib
 Requires-Dist: motmetrics
 Requires-Dist: ninja
 Requires-Dist: numpy
-Requires-Dist: numpy-indexed
+Requires-Dist: numpy_indexed
 Requires-Dist: opencv-contrib-python
 Requires-Dist: opencv-python
 Requires-Dist: pandas
@@ -52,4 +51,13 @@ Requires-Dist: thop
 Requires-Dist: tqdm
 Requires-Dist: vidgear
 Requires-Dist: yacs
-
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

{dnt-0.3.1.3.dist-info → dnt-0.3.1.7.dist-info}/RECORD

@@ -1,33 +1,34 @@
-dnt/__init__.py,sha256=Iz_w68hpgOZlVbPBv4li4b1BWVEEBQ2RlyuW84qtUnc,162
-dnt/analysis/__init__.py,sha256=wxNYOtG1cxw45WIEmS-Xlc71b5GkIQM6YxRn3JYvs4Y,161
-dnt/analysis/count.py,sha256=LCh3MvqnTivd_yQtucifyYYUeqxwIxqs8B60g-xBuNk,3806
-dnt/analysis/interaction.py,sha256=-KYMTfAvCKSuk2IKsIqkpKkF4ZzrQcGhxsbdYuUtdIw,26570
+dnt/__init__.py,sha256=z1Zd_a6E0VmW-afb-LTKnmchJdQg-jZ8vSJW_y6nvro,162
+dnt/analysis/__init__.py,sha256=-TzIL_MSIG76yGbxoAesaA8oXMLUtHaK7G92FOMnRTc,159
+dnt/analysis/count.py,sha256=6x7GvveygGOkRqgnzVD9lGaIlJN3HPprJbNkl7CdN18,4628
+dnt/analysis/interaction2.py,sha256=-c3yxIIDYHv8uCS653_HKxT2T02OP8AKqDwKLPpO6SE,26904
 dnt/analysis/position.py,sha256=RHCDv_R-2MsDDjy-sx3-qoY215NiflrgJ20cJVrbVBQ,239
 dnt/analysis/speed.py,sha256=KHHjompo4wT8Ft36dAQWq1yyJFJ89vVcpFJVaoz37-k,560
 dnt/analysis/stop.py,sha256=jIQkMnSJuIMK6_wQacVXLWtuP_JAqy7-5KyKhIAz8hs,10956
 dnt/analysis/stop2.py,sha256=UtRqSq3vT7BDKfeDD2b_LhNMSbPcafiLrLHDYxtkOlw,11128
-dnt/analysis/stop3.py,sha256=zwJx6ghJeNQ2arNjnI55zoKupdHOdhDPpWVcI_ov07A,38357
+dnt/analysis/stop3.py,sha256=TQB1dCTeXqCCn_oMUHvD2hejwrmtO0EEs-Q9hOkx0y8,38478
 dnt/detect/__init__.py,sha256=HEABuLXssGAlf1XB8GAo4DIq-e672Ls7_E0J9JL3CfA,110
-dnt/detect/signal/detector.py,sha256=wDEnXhb2f7V3YAd4qlZ6LUUZFJqSn3gn1_Rmw2DXWN0,11715
+dnt/detect/timestamp.py,sha256=qSUvaPs3OSqfzMynu95er0XNOplBZtNMS9P4eSHFwZs,3653
+dnt/detect/signal/detector.py,sha256=TZH-bpO4IunqJmxXs4Z8XtCg5s4CD9jVudd57efn-OU,11967
 dnt/detect/yolov8/__init__.py,sha256=r57LNAlmn1lNEcOEeVq1nye5AD6qVYp82XN9z0r3ZEY,95
-dnt/detect/yolov8/detector.py,sha256=XUO3AZvdRk5t_MRpdNmYc5T7d1767hgJwBd9ZbSKB34,10521
-dnt/detect/yolov8/segmentor.py,sha256=0tCFmqsrPaOyTFjGNB7S0RF15W9w8Q-AfqunNPg4-bU,4031
+dnt/detect/yolov8/detector.py,sha256=Yy9NkGUrsNgqdxWAZYRsYMg_lYD7uh9LeGPSX9DMEo4,12087
+dnt/detect/yolov8/segmentor.py,sha256=gPw5lbhiqoCUHs_IsRSoSslm0SPNmg3Ga4G04PQuXLI,6423
 dnt/engine/__init__.py,sha256=d_SI1PXnh_G2voYg6lxMQMwXPTIiYI7h5DbJXR5M8Sw,212
 dnt/engine/bbox_interp.py,sha256=I-rCPr36x5c8Uuikmml5SwV7gDQRlU1AAX68e8hoouA,3076
 dnt/engine/bbox_iou.py,sha256=5yQbPdXklliLe6WuB9U4wifehwoU41mU03x03R4jUoY,506
 dnt/engine/cluster.py,sha256=TWt_Rt0Q3mBWxWbTW_d5GpMOYVyVF5gWso5BmVBaZIU,822
 dnt/engine/iob.py,sha256=kwlwnYKYpzoCwgRjWkuPFXukJ7wZ1iUb6EXNzCIPUDQ,2209
 dnt/filter/__init__.py,sha256=xjvrrcNPtgoX7sSqXBzWk5RfPgFpUohhIXeYEXW-7mQ,99
-dnt/filter/filter.py,sha256=g7agZPYxvwgdSP1g0DAjf6GKQ19vUmoDbIq_cIdNh3A,20107
+dnt/filter/filter.py,sha256=0g4h-jwvkBl3zVNZX_bPsj4acb0wivsxdUhH7Q8NFWc,20528
 dnt/label/__init__.py,sha256=iTyqt8OlxIeRY6_mDKbOcEYSKYyoeEflEznF1ChpKKo,102
 dnt/label/labeler.py,sha256=M1rSzXNoohs6a7GgcuoG2LCNItpcGF9ba9FUUCdItx8,22831
-dnt/label/labeler2.py,sha256=sd1sXgbC1pj13S4kbA0BuG5_EZrIECl8o3hjrlEW1fU,21018
+dnt/label/labeler2.py,sha256=xD8ICHnBgbNPv8dc3gMy124N4WfbSerJsKUOnm1wWJc,24642
 dnt/shared/__init__.py,sha256=_1OvnXJdeCFtCgnXo4lVIOXI7ZUcJO8JjnQsqmu3oaM,98
 dnt/shared/convert.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dnt/shared/download.py,sha256=S_yVXrIh1bEpypYA0KuMovGy5ofO2-l2VnzWNgTpC_s,395
 dnt/shared/files.py,sha256=EwQl2tqyMP15E_lTnso1GJewvKvDBtUi_hx-75sBB5U,642
 dnt/shared/filter.py,sha256=fYvKCgrs5sTz1sQCK-jAGmEO-qYAjx1aTW9X2ULTu1A,1372
-dnt/shared/synhcro.py,sha256=oibZd4v3nC7APK3jro11pxWDVMJQlYlAcTh5OFngNRk,5932
+dnt/shared/synhcro.py,sha256=hKgaQ2gX9RTmW_GlMevAKtb3H69wytuYWvpBT24uIQE,5939
 dnt/shared/util.py,sha256=K57i2yQmtihPu_yhIhfrxUeKLy80d3Npcf0fMzR_JSQ,19446
 dnt/shared/data/coco.names,sha256=Y0oRMusz-AkdYPLDRqur6LkFrgg4cDeu2IOVO3Mpr4Q,625
 dnt/shared/data/openimages.names,sha256=RM3TTr5bRkQLbsXQl48pLe97V8fOoEnz1P557YjUBmQ,5476
@@ -271,8 +272,8 @@ dnt/third_party/fast-reid/fastreid/utils/timer.py,sha256=umKqc9hGOt5LgJucaDBhTkP
 dnt/third_party/fast-reid/fastreid/utils/visualizer.py,sha256=qNsZwsBKxXj3sIANr9RzB5maGVOlpH8VrkIrL2CX3uU,11831
 dnt/track/__init__.py,sha256=cmDf8vJxrE3kS5pDwzOROdM07wrRNf5865Rgac5n4q4,160
 dnt/track/config.py,sha256=8kd9bmJpEfn7U_y2LGLzL8BejVcQojArMPPGF-Y-K6o,6244
-dnt/track/re_class.py,sha256=8dG9axWS5mW4Yf7eJ7IEsSQHXlFiVuElh9l56CUaQD4,3614
-dnt/track/tracker.py,sha256=e4k7jhcyCO7wxIMNnqPn_UIOS30PLDpLFmBvQZpYsJo,10748
+dnt/track/re_class.py,sha256=SalCzJMNj-rKssc0iRhNHzokH5k_dJtb4GrCl6rOunw,4488
+dnt/track/tracker.py,sha256=gAvWdZe30ML13uZyNGfu9bH3J4nFEVNFmnJE9RzcA3E,10801
 dnt/track/botsort/__init__.py,sha256=9FB0CiokYiLbebof1l6M9lZQgFpLJ0awPRin8cCpi9A,100
 dnt/track/botsort/inference.py,sha256=-1o-KUQIAfgROI5c7iItRo--Ckfx_6eCfCexlP9eSmE,3254
 dnt/track/botsort/bot_tracker/__init__.py,sha256=DVRNfpNUsRw71bjP-ewihmhmmVSqjRexaTJ9V5vaJZw,124
@@ -286,29 +287,29 @@ dnt/track/botsort/bot_tracker/tracking_utils/evaluation.py,sha256=PShRltoCU3VnaK
 dnt/track/botsort/bot_tracker/tracking_utils/io.py,sha256=KtcgFGtTNpfRMT4g1lc2ZAON00wCWfTH5zgDLpwrg3k,3627
 dnt/track/botsort/bot_tracker/tracking_utils/timer.py,sha256=wjOo45QtJzTwtWt584Xaq8wBpXB2SclpQ6UYuVFpXSU,958
 dnt/track/dsort/__init__.py,sha256=osS5iI6LXf0A8ClpVB3x5LfI5eFYzkuT34X_PIXmuhI,97
-dnt/track/dsort/dsort.py,sha256=xUFZUcsBgrUPzAdcPzw9mnqnYvnnlstzfK7HcOJrn10,2776
+dnt/track/dsort/dsort.py,sha256=uwJgQ2SnyiUqnnCVJUXqZILuvrY9Dh_S9rfbWK1IHPs,3425
 dnt/track/dsort/configs/bagtricks_R50.yml,sha256=HI3gJPAf3FWcsZJenqYXv3xrMJQuWH85j2Xmj5fAnQg,134
 dnt/track/dsort/configs/deep_sort.yaml,sha256=RlWX74HFvoXmt-m7YbJyFID4qgI809ysK4g7ahQ9mJA,178
 dnt/track/dsort/configs/fastreid.yaml,sha256=7liz49xIZNDAalBawJg6xAch1DsRyRJWsSWSaVJf_ZE,97
 dnt/track/dsort/deep_sort/__init__.py,sha256=OP0dTJjoHrfIl9zxtWU0AIHlsz2iEaWD3kQL4tfgmWI,574
-dnt/track/dsort/deep_sort/deep_sort.py,sha256=OFt-9Dpy2Iy0JG_W95vKJ_CXEoBDgTVFGZu7eI06F_w,5225
+dnt/track/dsort/deep_sort/deep_sort.py,sha256=IxKF9CZxc6DAqyxM8tsHz9-Lerj0NO-G3zng6MT-ZiQ,5290
 dnt/track/dsort/deep_sort/deep/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dnt/track/dsort/deep_sort/deep/feature_extractor.py,sha256=JSSk2MjkzLo4WUKxuKWkUxKY3KrYnGf3iuB1iTKw8SU,5427
 dnt/track/dsort/deep_sort/deep/model.py,sha256=-OYmUtAhnRQofwp1Ar1p6Cl24j2ANokHcWvHYfQyqMs,3314
 dnt/track/dsort/deep_sort/deep/checkpoint/ckpt.t7,sha256=33Xd70LD0b2me8lLCT585h3n91qJ82qPhopChGIZgxY,46034619
 dnt/track/dsort/deep_sort/sort/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dnt/track/dsort/deep_sort/sort/detection.py,sha256=I_3hF5A7wE0RNAACi-lQjOrzVJeCSpZZYu1nK-UfyII,1474
+dnt/track/dsort/deep_sort/sort/detection.py,sha256=bV2AQVecMAP1GMdaqyo8Xqa87lMp9rfYj9ZYhCzBiB4,1503
 dnt/track/dsort/deep_sort/sort/iou_matching.py,sha256=X6MFCzpA3-5X6cTgV8lhgUzO6UtHGPzowfhxeWN8UIE,2907
 dnt/track/dsort/deep_sort/sort/kalman_filter.py,sha256=jOiYGSo9wQPcdPFvVAbxeMuMWlhTKp0cdbs2ymk2d5Y,8016
 dnt/track/dsort/deep_sort/sort/linear_assignment.py,sha256=nj9UwqrQYjUjnn0I5OTYISWfixi2CngPHdK0EEbKKQ8,8080
 dnt/track/dsort/deep_sort/sort/nn_matching.py,sha256=En6DgChB03Pa1HHJaiA9SxCmkfglUJz5cmoMdPmbHj0,5827
 dnt/track/dsort/deep_sort/sort/preprocessing.py,sha256=AOgEvA5OXouaGZnQiRbSzKUjOdo9zwxjClm4JAESG3M,1987
-dnt/track/dsort/deep_sort/sort/track.py,sha256=vqPzxPHtvDjgsPEJIpolQOObZ91I0Z2woSXzHeVi4GA,5142
-dnt/track/dsort/deep_sort/sort/tracker.py,sha256=RXl0xeRFUaBTgi0dfBzJ0DpAptjsxFM0vONC9J9SaZ4,5429
+dnt/track/dsort/deep_sort/sort/track.py,sha256=E_pd78wG_13myFmJAq73VXlRN3cnZ_jcxI3OHixyTZg,5176
+dnt/track/dsort/deep_sort/sort/tracker.py,sha256=Yo6WjefoHm6VGvr-LIKGNzTdcqocKGePcLogzRr2pU4,5444
 dnt/track/sort/__init__.py,sha256=3pUesXuRtu28cZQ3ORXBJtZ-stbWBrDyRSIC0IvmsDc,23
-dnt/track/sort/sort.py,sha256=6hMLSPYXXCTbxC1jkDHAj0ntw-gxeGH0zlFLK77dIhI,9753
-dnt-0.3.1.3.dist-info/LICENSE,sha256=ogjvEhk5GMZcGK_ykzVanElTHgnFdGANoG57hexoiVU,1067
-dnt-0.3.1.3.dist-info/METADATA,sha256=XpT-mxl_snTqbBsjMWOrB7tA-WobDQ4aVCeycsfaE5I,1450
-dnt-0.3.1.3.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-dnt-0.3.1.3.dist-info/top_level.txt,sha256=h1DZPTWPJuWyVPN837zOMtsfMP3Tl-rE-YumDuDahcU,4
-dnt-0.3.1.3.dist-info/RECORD,,
+dnt/track/sort/sort.py,sha256=zevTgan4EoBlxgoQGJTrQCDeKFVdCp7CSDhweMJrFLI,9714
+dnt-0.3.1.7.dist-info/licenses/LICENSE,sha256=ogjvEhk5GMZcGK_ykzVanElTHgnFdGANoG57hexoiVU,1067
+dnt-0.3.1.7.dist-info/METADATA,sha256=x6GRflO7_dFU_2faLOoJ5OobXEVhTm-3W46R_bgZwVM,1623
+dnt-0.3.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dnt-0.3.1.7.dist-info/top_level.txt,sha256=h1DZPTWPJuWyVPN837zOMtsfMP3Tl-rE-YumDuDahcU,4
+dnt-0.3.1.7.dist-info/RECORD,,

{dnt-0.3.1.3.dist-info → dnt-0.3.1.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (74.1.2)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any