dnt 0.2.4__tar.gz → 0.3.1.7__tar.gz

dnt-0.3.1.7/MANIFEST.in

@@ -0,0 +1,6 @@
+recursive-include dnt *.py
+recursive-include dnt *.yaml
+recursive-include dnt *.yml
+recursive-include dnt *.names
+recursive-include dnt *.t7
+recursive-exclude __pycache__ *

{dnt-0.2.4/dnt.egg-info → dnt-0.3.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: dnt
-Version: 0.2.4
+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
@@ -13,24 +13,51 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
 License-File: LICENSE
-Requires-Dist: numpy
-Requires-Dist: pandas
-Requires-Dist: matplotlib
 Requires-Dist: torch
 Requires-Dist: torchvision
 Requires-Dist: torchaudio
-Requires-Dist: ultralytics
+Requires-Dist: ultralytics>=8.2
+Requires-Dist: cython
+Requires-Dist: cython_bbox
 Requires-Dist: easydict
+Requires-Dist: faiss-cpu
 Requires-Dist: filterpy
+Requires-Dist: gdown
+Requires-Dist: geopandas
+Requires-Dist: h5py
+Requires-Dist: lapx
+Requires-Dist: loguru
 Requires-Dist: matplotlib
+Requires-Dist: motmetrics
+Requires-Dist: ninja
+Requires-Dist: numpy
 Requires-Dist: numpy_indexed
-Requires-Dist: opencv-python
 Requires-Dist: opencv-contrib-python
+Requires-Dist: opencv-python
 Requires-Dist: pandas
-Requires-Dist: geopandas
-Requires-Dist: shapely
-Requires-Dist: scipy
-Requires-Dist: scikit-learn
+Requires-Dist: Pillow
+Requires-Dist: prettytable
+Requires-Dist: pycocotools
+Requires-Dist: pyyaml
 Requires-Dist: scikit-image
+Requires-Dist: scikit-learn
+Requires-Dist: scipy
+Requires-Dist: shapely
+Requires-Dist: sympy
+Requires-Dist: tabulate
+Requires-Dist: tensorboard
+Requires-Dist: termcolor
+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.2.4 → dnt-0.3.1.7}/README.md

@@ -15,8 +15,13 @@ This tool can generate vehicel trajectories using real-time object detection and
 * Libarary: pytorch (>=1.12 with cuda), pandas, tqdm, numpy, scipy, matplotlib, ultralytics
 
 ### Installing
-
 ```
+Setup enviroment:
+
+    python3.x -m venv dnt (you may replace dnt with you specified name; python version >3.9)
+    sudo apt-get install python3.x-dev  
+    pip install python-dev-tools
+
 pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
 pip install dnt
 ```

dnt-0.3.1.7/dnt/__init__.py

@@ -0,0 +1,5 @@
+import sys, os
+sys.path.append(os.path.dirname(__file__))
+sys.path.append(os.path.join(os.path.dirname(__file__), 'third_party/fast-reid'))
+
+__version__='0.3.1.7'

dnt-0.3.1.7/dnt/analysis/__init__.py

@@ -0,0 +1,8 @@
+import os
+import sys
+
+sys.path.append(os.path.dirname(__file__))
+
+from stop3 import StopAnalyzer
+from count import Count
+from interaction2 import YieldAnalyzer

dnt-0.3.1.7/dnt/analysis/count.py

@@ -0,0 +1,107 @@
+# pylint: disable=too-many-arguments
+'''
+This script is used to count the number of tracks that passing a reference line for post-analysis
+last modified: 2021-09-30
+'''
+from shapely.geometry import Polygon, LineString
+import geopandas as gpd
+import pandas as pd
+from tqdm import tqdm
+import numpy as np
+
+class Count:
+    '''
+    Count tracks that passing a reference line for post-analysis
+    Methods:
+        __init__(): initialize the class
+        count_tracks_by_line(): 
+            count tracks that passing a reference line for post-analysis
+    '''
+    def __init__(self) -> None:
+        self.track_fields = ['frame', 'track_id', 'x', 'y', 'w', 'h', 'r1', 'r2', 'r3', 'r4']
+
+    def count_tracks_by_line(self,
+                             tracks:pd.DataFrame=None,
+                             track_file:str=None,
+                             line: LineString=None,
+                             video_index:int=None,
+                             video_tot:int=None,
+                             verbose:bool=True) -> pd.DataFrame:
+        '''
+        Count tracks that passing a reference line for post-analysis
+            Inputs:
+                tracks - a DataFrame of tracks, if None (default), read track_file
+                track_file -  a txt file contains tracks, 
+                line - the reference line, LineString(pointA, pointB)
+                video_index - the index of video for processing
+                video_tot - the total number of videos
+                verbose - if True, show progress bar
+            Return:
+                A DataFrame of track_id, frame, direction, count
+        '''
+        # Load tracks
+        if tracks is None:
+            tracks = pd.read_csv(track_file, header=None)
+        if len(tracks.columns) != len(self.track_fields):
+            raise Exception("The tracks format is incorrect!")
+        tracks.columns = self.track_fields
+        track_ids = tracks['track_id'].unique()
+
+        # Create a GeoDataFrame of tracks
+        geo = tracks.apply(lambda track: Polygon([(track['x'], track['y']),
+                                                (track['x'] + track['w'], track['y']),
+                                                (track['x'] + track['w'], track['y'] + track['h']),
+                                                (track['x'], track['y'] + track['h'])]),
+                                                axis =1)
+        geo_tracks = gpd.GeoDataFrame(tracks, geometry=geo)
+        point_a = np.array(line.coords[0])
+        point_b = np.array(line.coords[1])
+
+        # Interate through all tracks
+        if verbose:
+            pbar = tqdm(total=len(track_ids), unit=' tracks')
+            if video_index and video_tot:
+                pbar.set_description_str("Counting {} of {}".format(video_index, video_tot))
+            else:
+                pbar.set_description_str("Counting")
+        intersected_tracks = []
+        intersected_frames = []
+        intersected_direct = []
+
+        for track_id in track_ids:
+            selected = geo_tracks.loc[(geo_tracks['track_id']==track_id)].copy()
+            if len(selected)>0:
+                selected['intersected'] = line.intersects(selected.geometry).values.tolist()
+                intersected = selected.loc[(selected['intersected']==True)].copy()
+                if len(intersected) > 0:
+                    intersected.sort_values(by='frame', inplace=True)
+                    if not track_id in intersected_tracks:
+                        intersected_tracks.append(track_id)
+                        frame_pos = int(len(intersected)/2)
+                        intersected_frames.append(intersected['frame'].values[frame_pos])
+
+                        # center point of the first frame
+                        point_c = np.array((intersected.iloc[0]['x'] + intersected.iloc[0]['w']/2,
+                                    intersected.iloc[0]['y'] + intersected.iloc[0]['h']/2))
+                        d = 2  # right
+                        if np.cross(point_c-point_a, point_b-point_a) < 0:
+                            d = 1 # left
+                        intersected_direct.append(d)
+            if verbose:
+                pbar.update()
+                
+        if verbose:
+            pbar.close()
+
+        results = pd.DataFrame(
+            {
+                'track_id': intersected_tracks,
+                'frame': intersected_frames,
+                'direction': intersected_direct,
+            }
+        )
+        results.sort_values(by='frame', inplace=True)
+        results = results.reset_index(drop=True)
+        results['count'] = results.index + 1
+
+        return results