dnt 0.3.2.2__tar.gz → 0.3.2.4__tar.gz

{dnt-0.3.2.2/dnt.egg-info → dnt-0.3.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dnt
-Version: 0.3.2.2
+Version: 0.3.2.4
 Summary: A Python tool for video-based traffic analytiscs
 Author-email: Zhenyu Wang <wonstran@hotmail.com>
 Project-URL: Homepage, https://github.com/wonstran/dnt

{dnt-0.3.2.2 → dnt-0.3.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dnt"
-version = "0.3.2.2"
+version = "0.3.2.4"
 description = "A Python tool for video-based traffic analytiscs"
 readme = "README.md"
 requires-python = ">=3.9"
@@ -81,8 +81,11 @@ Homepage = "https://github.com/wonstran/dnt"
 [tool.setuptools]
 include-package-data = true
 
+[tool.setuptools.package-dir]
+"" = "src"
+
 [tool.setuptools.packages.find]
-where = ["."]
+where = ["src"]
 
 [tool.ruff]
 line-length = 100

{dnt-0.3.2.2 → dnt-0.3.2.4}/src/dnt/__init__.py

@@ -8,4 +8,4 @@ import sys
 
 sys.path.append(os.path.dirname(__file__))
 
-__version__ = "0.3.2.3"
+__version__ = "0.3.2.4"

{dnt-0.3.2.2 → dnt-0.3.2.4}/src/dnt/detect/yolo/detector.py

@@ -213,7 +213,7 @@ class Detector:
         end_frame: int | None = None,
         verbose: bool = True,
         show: bool = False,
-        disp_filename: bool = False,
+        message: str | None = None,
     ) -> pd.DataFrame:
         """Run object detection on a video and return per-frame detections.
 
@@ -235,8 +235,9 @@ class Detector:
             Whether to show a progress bar. Default is True.
         show : bool, optional
             Whether to display the video frames with detections. Default is False.
-        disp_filename: bool, optional
-            Whether to show the file name in the progress bar. Default is False.
+        message : str | None, optional
+            Optional message shown in the progress bar description.
+            Default is None.
 
         Returns
         -------
@@ -270,21 +271,18 @@ class Detector:
 
         # Some codecs return 0 or -1 for frame count
         if verbose:
-            if tot_frames <= 0:
-                pbar = tqdm(desc="Detecting", unit="frame")
-            else:
-                pbar = tqdm(total=frame_total, desc="Detecting", unit="frame")
-
             if (video_index is not None) and (video_tot is not None):
                 desc = f"Detecting {video_index} of {video_tot}"
-                if disp_filename:
-                    desc += f" - {Path(input_video).name}"
-                    pbar.set_description_str(desc)
             else:
                 desc = "Detecting"
-                if disp_filename:
-                    desc += f" {Path(input_video).name}"
-                    pbar.set_description_str(desc)
+
+            if message is not None:
+                desc += f" {message}"
+
+            if tot_frames <= 0:
+                pbar = tqdm(desc=desc, unit="frame")
+            else:
+                pbar = tqdm(total=frame_total, desc=desc, unit="frame")
 
         results: list[dict] = []
         frame_idx = start_frame
@@ -508,7 +506,7 @@ class Detector:
         is_overwrite: bool = False,
         is_report: bool = True,
         verbose: bool = True,
-        disp_filename: bool = True,
+        message: str | None = None,
     ) -> list[str]:
         """Run detection on multiple videos and optionally write per-video output files.
 
@@ -527,8 +525,9 @@ class Detector:
             included in the returned list.
         verbose : bool, optional
             If True, prints progress messages. Default is True.
-        disp_filename: bool, optional
-            If True, prints file name in progress bar. Default is True.
+        message : str | None, optional
+            Optional message shown in each progress bar description.
+            Default is None.
 
         Returns
         -------
@@ -564,7 +563,7 @@ class Detector:
                 video_index=idx,
                 video_tot=total_videos,
                 verbose=verbose,
-                disp_filename=disp_filename,
+                message=message,
             )
 
             if iou_file is not None:

{dnt-0.3.2.2 → dnt-0.3.2.4}/src/dnt/filter/filter.py

@@ -180,18 +180,21 @@ class Filter:
         video_index: int | None = None,
         video_tot: int | None = None,
     ) -> pd.DataFrame:
-        """Filter tracks by zones.
+        """Filter tracks by polygon zones using an aggregate ID match.
+
         Inputs:
-            tracks: tracks
-            zones: a list of polygons
-            method - 'include' (default), 'exclude'
-            ref_point - the reference point of a track bbox,
-                defalt is br, others - bl, bc, tl, tc, tr, cl, cc, cr
-            offset - the offset to ref_point, default is (0, 0)
-            video_index - video index
-            video_tot - total videos
-        Return:
-            Filtered tracks
+            tracks - input tracks.
+            zones - polygons used for matching.
+            method - 'include' (default) or 'exclude'.
+            ref_point - bbox reference point (default: 'bc'); options include
+                bl, bc, br, tl, tc, tr, cl, cc, cr.
+            offset - offset applied to the reference point, default is (0, 0).
+            video_index - video index for progress display.
+            video_tot - total video count for progress display.
+
+        Returns:
+            Filtered tracks.
+
         """
         if offset is None:
             offset = (0, 0)
@@ -299,17 +302,20 @@ class Filter:
         video_index: int | None = None,
         video_tot: int | None = None,
     ) -> pd.DataFrame:
-        """Filter tracks by zones.
+        """Filter frame rows by polygon zones.
+
         Inputs:
-            tracks - tracks
-            zones - zones (polygon)
-            method - 'include' (default) include the tracks if they are within the zones; 'exclude' exclude the tracks if they are within the zones
-            ref_point - the reference point of a track bbox, defalt is bottom_point, center_point, left_up, right_up, left_buttom, right_buttom
-            offset - the offset to ref_point, default is (0, 0)
-            video_index - video index
-            video_tot - total videos
-        Return:
-            Filtered tracks
+            tracks - input tracks.
+            zones - polygons used for matching.
+            method - 'include' (default) keeps matching rows; 'exclude' drops them.
+            ref_point - bbox reference point (default: 'bc').
+            offset - offset applied to the reference point, default is (0, 0).
+            video_index - video index for progress display.
+            video_tot - total video count for progress display.
+
+        Returns:
+            Filtered frame rows.
+
         """
         if offset is None:
             offset = (0, 0)
@@ -427,7 +433,7 @@ class Filter:
                      'filter' - filter tracks within zones
                      'label' - label tracks with zone index
             ref_point - the reference point of a track bbox,
-                        br - buttom_right,
+                        br - bottom_right,
                         bl - bottom_left
                         bc - bottom_center
                         cc - center_point,
@@ -437,11 +443,10 @@ class Filter:
                         tl - top_left,
                         tr - top_right,
             offset - the offset to ref_point, default is (0, 0)
-            aggregate - combine outputs to one dataframe, add zone column
-            zone_name - if aggregate, the field name of zone variable, default is 'zone'
+            zone_name - field name used when method='label', default is 'zone'
             video_index - video index
             video_tot - total videos
-        Return:
+        Returns:
             Filtered tracks
         """
         if offset is None:
@@ -538,7 +543,7 @@ class Filter:
                 the lines, exclude - exclude tracks crossing the lines
             video_index - the index of video for processing
             video_tot - the total number of videos
-        Return:
+        Returns:
             a DataFrame of [FRAME, TRACK_ID, TOPX, TOPY, WIDTH, LENGTH, RESERVED, RESERVED, RESERVED]
         """
         track_ids = tracks[1].unique()
@@ -601,13 +606,13 @@ class Filter:
             lines - a list of LineString
             method - filtering method, include (default) - including tracks crossing the
                 lines, exclude - exclude tracks crossing the lines
-            tolerance - if a bbox intesect the reference lines of (number of lanes -
+            tolerance - if a bbox intersects the reference lines of (number of lanes -
                 tolerance), it is hit. default is 0.
             force_line_indexes: the line indexes that a bbox must intersect for matching
-            bbox_size - the size of detection bbox, default is 0 - the orginal bbox
+            bbox_size - the size of detection bbox, default is 0 - the original bbox
             video_index - the index of video for processing
             video_tot - the total number of videos
-        Return:
+        Returns:
             a DataFrame of [FRAME, TRACK_ID, TOPX, TOPY, WIDTH, LENGTH, RESERVED,
                 RESERVED, RESERVED]
         """

{dnt-0.3.2.2 → dnt-0.3.2.4}/src/dnt/track/__init__.py

@@ -21,7 +21,7 @@ from .tracker import (
     BoostTrackConfig as BoostTrackConfig,
 )
 from .tracker import (
-    BoTSORTConfig as BOTSORTConfig,
+    BoTSORTConfig as BoTSORTConfig,
 )
 from .tracker import (
     ByteTrackConfig as ByteTrackConfig,
@@ -54,8 +54,12 @@ from .tracker import (
     Tracker as Tracker,
 )
 
+# Backward-compat aliases for historical exported names.
+BOTSORTConfig = BoTSORTConfig
+SFSortConfig = SFSORTConfig
+
 __all__ = [
-    "BOTSORTConfig",
+    "BoTSORTConfig",
     "BoostTrackConfig",
     "ByteTrackConfig",
     "DeepOCSORTConfig",
@@ -66,7 +70,8 @@ __all__ = [
     "ReClass",
     "ReIDWeights",
     "SFSORTConfig",
-    "SFSortConfig",
     "StrongSORTConfig",
     "Tracker",
+    "interpolate_tracks_rts",
+    "link_tracklets",
 ]

{dnt-0.3.2.2 → dnt-0.3.2.4}/src/dnt/track/tracker.py

@@ -690,15 +690,16 @@ class Tracker:
         video_file: str | None = None,
         show: bool = False,
         video_index: int | None = None,
-        total_videos: int | None = None,
+        video_tot: int | None = None,
+        message: str | None = None,
     ) -> pd.DataFrame:
         """Run tracking on a single detection file using BoxMOT.
 
         Parameters
         ----------
         det_file : str
-            Path to detection file (CSV format with columns:
-            frame, x1, y1, width, height, confidence, class_id).
+            Path to detection file in DNT detection format
+            (frame, -, x, y, width, height, confidence, class_id).
         out_file : str
             Path to write tracking results. If empty string, results are not saved.
         video_file : str, optional
@@ -709,8 +710,11 @@ class Tracker:
             'q' to stop tracking early.
         video_index : int, optional
             Index of current video in batch (for progress bar display).
-        total_videos : int, optional
+        video_tot : int, optional
             Total number of videos in batch (for complete progress context).
+        message : str | None, optional
+            Optional progress text shown in the progress bar. If None, the
+            video stem is used (default: None).
 
         Returns
         -------
@@ -752,7 +756,8 @@ class Tracker:
             out_file=out_file,
             show=show,
             video_index=video_index,
-            total_videos=total_videos,
+            video_tot=video_tot,
+            message=message,
         )
 
     def track_batch(
@@ -762,6 +767,7 @@ class Tracker:
         output_path: str | None = None,
         is_overwrite: bool = False,
         is_report: bool = True,
+        message: str | None = None,
     ) -> list[str]:
         """Run tracking on multiple detection files sequentially.
 
@@ -781,6 +787,9 @@ class Tracker:
             If False (default), skip tracking for videos with existing output files.
         is_report : bool, optional
             If True (default), include skipped files in returned list.
+        message : str | None, optional
+            Optional progress text shown in each tracking progress bar.
+            If None (default), each video's stem is used.
 
         Returns
         -------
@@ -829,7 +838,8 @@ class Tracker:
                 out_file=track_file if track_file else "",  # track() expects a path
                 video_file=video_file,
                 video_index=idx,
-                total_videos=total_videos,
+                video_tot=total_videos,
+                message=message,
             )
 
             if track_file:
@@ -1083,7 +1093,8 @@ class Tracker:
         out_file: str,
         show: bool = False,
         video_index: int | None = None,
-        total_videos: int | None = None,
+        video_tot: int | None = None,
+        message: str | None = None,
     ) -> pd.DataFrame:
         """Execute BoxMOT tracking on one video with corresponding detections.
 
@@ -1103,8 +1114,11 @@ class Tracker:
             If True, display tracking preview (default: False).
         video_index : int | None, optional
             Index in batch progression (for UI display).
-        total_videos : int | None, optional
+        video_tot : int | None, optional
             Total videos in batch (for UI display).
+        message : str | None, optional
+            Optional progress text shown in the progress bar. If None, the
+            video stem is used.
 
         Returns
         -------
@@ -1112,7 +1126,6 @@ class Tracker:
             Tracking results with TRACK_FIELDS columns.
 
         """
-        """Run BoxMOT tracking on one video/detection pair and return track records."""
         detections = pd.read_csv(det_file, header=None).to_numpy()
         if detections.size == 0 or len(detections) == 0:
             return pd.DataFrame(columns=self.TRACK_FIELDS)
@@ -1134,10 +1147,11 @@ class Tracker:
             half=self.half,
         )
 
-        desc = f"Tracking {Path(video_file).stem}"
+        pbar_message = message if message is not None else Path(video_file).stem
+        desc = f"Tracking {pbar_message}"
         pbar = tqdm(total=total_frames, desc=desc, unit="frames")
-        if video_index is not None and total_videos is not None:
-            pbar.set_description_str(f"Tracking {video_index} of {total_videos} - {Path(video_file).stem}")
+        if video_index is not None and video_tot is not None:
+            pbar.set_description_str(f"Tracking {video_index} of {video_tot} - {pbar_message}")
 
         results: list[list[float | int]] = []
         cap.set(cv2.CAP_PROP_POS_FRAMES, start_frame)

{dnt-0.3.2.2 → dnt-0.3.2.4/src/dnt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dnt
-Version: 0.3.2.2
+Version: 0.3.2.4
 Summary: A Python tool for video-based traffic analytiscs
 Author-email: Zhenyu Wang <wonstran@hotmail.com>
 Project-URL: Homepage, https://github.com/wonstran/dnt

{dnt-0.3.2.2 → dnt-0.3.2.4/src}/dnt.egg-info/SOURCES.txt

@@ -2,23 +2,12 @@ LICENSE.md
 MANIFEST.in
 README.md
 pyproject.toml
-dnt.egg-info/PKG-INFO
-dnt.egg-info/SOURCES.txt
-dnt.egg-info/dependency_links.txt
-dnt.egg-info/requires.txt
-dnt.egg-info/top_level.txt
-examples/example.py
-examples/example_batch.py
-examples/example_extract_frames.py
-examples/example_filter.py
-examples/example_label.py
-examples/example_post-tracklets.py
-examples/example_post.py
-examples/example_shapes.py
-examples/example_shapes_v2.py
-examples/example_stop.py
-examples/example_track.py
 src/dnt/__init__.py
+src/dnt.egg-info/PKG-INFO
+src/dnt.egg-info/SOURCES.txt
+src/dnt.egg-info/dependency_links.txt
+src/dnt.egg-info/requires.txt
+src/dnt.egg-info/top_level.txt
 src/dnt/detect/__init__.py
 src/dnt/detect/timestamp.py
 src/dnt/detect/signal/__init__.py

dnt-0.3.2.4/src/dnt.egg-info/top_level.txt

@@ -0,0 +1 @@
+dnt

dnt-0.3.2.2/dnt.egg-info/top_level.txt

@@ -1,6 +0,0 @@
-dist
-docs
-examples
-site
-src
-tests

dnt-0.3.2.2/examples/example.py

@@ -1,43 +0,0 @@
-import pathlib
-import sys
-
-root = pathlib.Path(__file__).resolve().parents[1]
-sys.path.insert(0, str(root / "src"))
-
-from dnt.detect import Detector  # noqa: E402
-from dnt.label.labeler import Labeler  # noqa: E402
-from dnt.track import Config, Tracker  # noqa: E402
-
-video_file = "/mnt/d/videos/sample/traffic.mp4"
-det_file = "/mnt/d/videos/sample/dets/traffic_det.txt"
-track_file = "/mnt/d/videos/sample/tracks/traffic_track.txt"
-label_file = "/mnt/d/videos/sample/labels/traffic.mp4"
-
-"""
-video_file = "/mnt/d/videos/samples/traffic_short.mp4"
-mot_file = "/mnt/d/videos/samples/traffic_short_mot.txt"
-iou_file = "/mnt/d/videos/samples/dets/traffic_short_iou.txt"
-iou_file_filtered = "/mnt/d/videos/samples/dets/traffic_short_iou_filtered.txt"
-track_file = "/mnt/d/videos/samples/tracks/traffic_short_track.txt"
-label_file = "/mnt/d/videos/samples/labels/traffic_short_track.mp4"
-label_file_2 = "/mnt/d/videos/samples/labels/traffic_short_det.mp4"
-"""
-detector = Detector()
-ious = detector.detect(video_file, iou_file=det_file, show_filename=True)
-# ious = detector.detect_frames(input_video=video_file, frames=list(range(1, 100)))
-
-# tracker = Tracker(cfg=Config.get_cfg_dsort())
-tracker = Tracker(cfg=Config.get_cfg_botsort())
-tracker.track(det_file, track_file, video_file)
-"""
-class_list = [0, 1, 2, 3, 5, 7]   
-filter_class_iou(iou_file, iou_file_filtered, class_list)
-
-tracker = Tracker(cfg=Config.get_cfg_botsort())
-tracker.track(iou_file_filtered, track_file, video_file, video_index=1, total_videos=1)
-"""
-labeler = Labeler()
-# labeler.draw_dets(input_video=video_file, output_video=label_file, det_file=det_file)
-labeler.draw_tracks(track_file=track_file, input_video=video_file, output_video=label_file, tail=100, class_name=True)
-
-print("ok")

dnt-0.3.2.2/examples/example_batch.py

@@ -1,42 +0,0 @@
-import sys
-from pathlib import Path
-
-# add project src/ to import path
-ROOT = Path(__file__).resolve().parents[1]
-sys.path.insert(0, str(ROOT / "src"))
-
-from dnt.detect import Detector  # noqa: E402
-
-video_path = Path("/mnt/d/videos/sample")
-det_path = Path("/mnt/d/videos/samples/dets")
-track_path = Path("/mnt/d/videos/samples/tracks")
-label_path = Path("/mnt/d/videos/samples/labels")
-
-# collect mp4s
-input_videos = list(video_path.glob("*.mp4"))
-
-# create output folder if it doesn't exist
-det_path.mkdir(parents=True, exist_ok=True)
-
-detector = Detector(half=True, weights="n")
-dets = detector.detect_batch(
-    [str(p) for p in input_videos],
-    output_path=str(det_path),
-    is_overwrite=True,
-)
-
-'''
-print('Tracking ...')
-cfg = Tracker.export_default_cfg("dsort")
-cfg['max_age'] = 5
-cfg['nn_budget'] = 30
-cfg['nms_max_overlap'] = 1
-
-tracker = Tracker(method='dsort', deepsort_cfg=cfg)
-tracks = tracker.track_batch(dets, input_videos, track_path, is_overwrite=True)
-
-print('Labeling ...')
-labeler = Labeler()
-labeler.draw_batch(tracks, input_videos, label_path, is_overwrite=True)
-'''
-print("ok")

dnt-0.3.2.2/examples/example_extract_frames.py

@@ -1,16 +0,0 @@
-import os
-
-from dnt.label.labeler import Labeler
-
-input_path = "/mnt/d/videos/benesch/24-120161 (Tampa)/Cam 1 (EB view)"
-out_path = "/mnt/d/videos/benesch/24-120161 (Tampa)/frames/eb"
-videos = os.listdir(input_path)
-
-labeler = Labeler()
-for video in videos:
-    labeler.export_frames(
-        input_video=os.path.join(input_path, video), frames=[0], output_path=out_path, prefix=os.path.splitext(video)[0]
-    )
-    print(os.path.join(input_path, video))
-
-print("okay")

dnt-0.3.2.2/examples/example_filter.py

@@ -1,65 +0,0 @@
-from dnt.analysis import StopAnalyzer
-from dnt.label import Labeler
-from dnt.detect import Detector
-from dnt.filter import Filter
-import pandas as pd
-from shapely import geometry
-from geopandas import gpd
-import datetime, os, tqdm
-
-
-lines = pd.read_csv("/mnt/d/videos/hfst_nb/frame/before_lines.csv")
-lines_v = lines.loc[lines['h']==0][['x1', 'y1', 'x2', 'y2']].values.tolist()
-lines_h = lines.loc[lines['h']==1][['x1', 'y1', 'x2', 'y2']].values.tolist()
-
-v_coords = []
-for line in lines_v:
-    v_coords.append([(line[0], line[1]), (line[2], line[3])])
-
-h_coords = []
-for line in lines_h:
-    h_coords.append([(line[0], line[1]), (line[2], line[3])])
-
-event_dict = {
-        0: [0],     # Stop
-        1: [1],     # Pass
-        2: [2]      # Invade
-    }
-
-rt_zones = [[(116, 209), (175, 204), (188, 332), (113, 340)]]
-
-video_file = "/mnt/d/videos/hfst_nb/labels/Standard_SCU7VM_2022-09-13_1600.003_lines.mp4"
-track_file = "/mnt/d/videos/hfst_nb/dets/Standard_SCU7VM_2022-09-13_1600.003_track.txt"
-track_file_filtered = "/mnt/d/videos/hfst_nb/dets/Standard_SCU7VM_2022-09-13_1600.003_track_filtered.txt"
-track_file2 = "/mnt/d/videos/hfst_nb/stops/Standard_SCU7VM_2022-09-13_1600.003_track2.txt"
-result_file = "/mnt/d/videos/hfst_nb/stops/Standard_SCU7VM_2022-09-13_1600.003_result.txt"
-label_file = "/mnt/d/videos/hfst_nb/stops/Standard_SCU7VM_2022-09-13_1600.003_label.txt"
-label_video = "/mnt/d/videos/hfst_nb/stops/Standard_SCU7VM_2022-09-13_1600.003_stop.mp4"
-
-init_time_str = "2022-09-13 6:00:01 PM"
-
-tracks = pd.read_csv(track_file, header=None)
-
-z = []
-for zone in rt_zones:
-    z.append(geometry.Polygon(zone))
-geo_zones = geometry.MultiPolygon(z)
-
-tracks = Filter.filter_tracks(tracks, exclude_zones=geo_zones)
-tracks.to_csv(track_file_filtered, header=None, index=False)
-
-analyzer = StopAnalyzer(h_coords=h_coords, v_coords=v_coords, event_dict=event_dict)
-analyzer.analysis(track_file=track_file_filtered, result_file=result_file, output_file=track_file2)
-results = pd.read_csv(result_file)
-results = results.loc[results['LANE']!=-1]
-results.to_csv(result_file, index=False)
-
-if init_time_str:
-    init_time=datetime.datetime.strptime(init_time_str, '%Y-%m-%d %I:%M:%S %p')
-    fps = Detector.get_fps(video_file)
-    StopAnalyzer.add_timestamp(result_file, result_file, init_time, fps)
-
-StopAnalyzer.export_label(track_file2, result_file, label_file, label_field=6, event_label=['STOP', 'PASS', 'INVADE'], vid_disp=False)
-
-labeler = Labeler(compress_message=False, label_fields=[6], zoom_factor=0.5)
-labeler.draw(label_file, video_file, label_video, video_index=1, video_tot=1)

dnt-0.3.2.2/examples/example_label.py

@@ -1,73 +0,0 @@
-"""Example script for labeling video tracks using the Labeler class."""
-
-import os
-import pathlib
-import sys
-
-root = pathlib.Path(__file__).resolve().parents[1]
-sys.path.insert(0, str(root / "src"))
-
-from dnt.label import Encoder, Labeler, LabelMethod, TrackClipMethod
-
-
-def main() -> None:
-    """Label video tracks using the Labeler class with FFMPEG method."""
-    input_video = "/mnt/d/videos/sample/traffic.mp4"
-    det_file = "/mnt/d/videos/sample/dets/traffic_det.txt"
-    track_file = "/mnt/d/videos/sample/tracks/traffic_track.txt"
-    track_label = "/mnt/d/videos/sample/tracks/traffic_label.txt"
-    det_video = "/mnt/d/videos/sample/labels/traffic_det.mp4"
-    track_video = "/mnt/d/videos/sample/labels/traffic_track.mp4"
-    clip_video = "/mnt/d/videos/sample/traffic_clip.mp4"
-
-    track_clip_output = "/mnt/d/videos/sample/labels/track_clips"
-    os.makedirs(track_clip_output, exist_ok=True)
-
-    labeler = Labeler(method=LabelMethod.CHROME_SAFE, encoder=Encoder.LIBX265)
-    """
-    labeler.draw_tracks(
-        input_video=input_video,
-        output_video=track_video,
-        track_file=track_file,
-        label_class_name=True,
-        thick=1,
-        size=0.8,
-    )
-    """
-    # labeler.draw_dets(input_video, det_video, det_file=det_file, label_score=True, thick=1, size=0.8)
-    shapes = [
-        {
-            "type": "polygon",
-            "geometry": [(200, 200), (400, 200), (400, 400)],
-            "fill": True,
-            "color": (255, 0, 0),
-            "alpha": 0.2,
-            "size": 20,
-            "thick": 1,
-        },
-    ]
-
-    labeler.draw_track_clips(
-        input_video=input_video,
-        output_path=track_clip_output,
-        track_file=track_file,
-        method=TrackClipMethod.RANDOM,
-        random_number=5,
-        tail_length=0,
-        label_class=True,
-        label_prefix=True,
-        thick=1,
-        size=0.8,
-        fill=True,
-        alpha=0.2,
-        color="random",
-    )
-
-    # labeler.clip_by_time(input_video=input_video, output_video=clip_video, clip_len_sec=5.0, method=LabelMethod.FFMPEG)
-    # frames = range(800,1000)
-    # labeler.export_frames(input_video, frames, "/mnt/d/videos/ped2stage/frames")
-    print("ok")
-
-
-if __name__ == "__main__":
-    main()

dnt-0.3.2.2/examples/example_post-tracklets.py

@@ -1,51 +0,0 @@
-"""Example script for object tracking using DNT library."""
-
-import os
-import sys
-from pathlib import Path
-
-# Allow running this script directly from the repository root.
-ROOT = Path(__file__).resolve().parents[1]
-SRC = ROOT / "src"
-if str(SRC) not in sys.path:
-    sys.path.insert(0, str(SRC))
-
-from dnt.detect import Detector, DetectorModel
-from dnt.label import Labeler, LabelMethod
-from dnt.label.labeler import Labeler, LabelMethod
-from dnt.track import BOTSORTConfig, ReIDWeights, Tracker, interpolate_tracks_rts, link_tracklets
-
-base_dir = "/mnt/e/videos/tmc/before/009/View_South/videos/4-19-23"
-input_video = f"{base_dir}/4-19-23-hiv00307.mp4"
-det_file = f"{base_dir}/dets/4-19-23-hiv00307_iou.txt"
-track_file = f"{base_dir}/tracks/4-19-23-hiv00307_ped_track_bot.txt"
-track_interp_file = f"{base_dir}/tracks/4-19-23-hiv00307_ped_track_bot_interp.txt"
-tracks_linked_file = f"{base_dir}/tracks/4-19-23-hiv00307_ped_track_bot_linked.txt"
-label_dir = f"{base_dir}/labels/track_clips"
-os.makedirs(label_dir, exist_ok=True)
-
-detector = Detector(model=DetectorModel.RTDETRx, device="auto")
-# detector.detect(input_video, det_file)
-
-cfg = BOTSORTConfig(ReIDWeights.CLIP_VEHICLEID)
-tracker = Tracker(cfg, device="auto")
-# tracks = tracker.track(det_file, track_file, input_video)
-
-labeler = Labeler(LabelMethod.CHROME_SAFE)
-
-labeler.draw_track_clips(
-    track_file=track_file, input_video=input_video, output_path=label_dir, label_prefix=False, fill=True, alpha=0.2
-)
-tracks_linked = link_tracklets(track_file=track_file, output_file=tracks_linked_file, max_gap=30)
-tracks_interp = interpolate_tracks_rts(track_file=track_file, output_file=track_interp_file, verbose=True)
-
-labeler.draw_track_clips(
-    track_file=tracks_linked_file,
-    input_video=input_video,
-    output_path=label_dir,
-    label_prefix=True,
-    fill=True,
-    alpha=0.2,
-)
-
-print("ok")

dnt-0.3.2.2/examples/example_post.py

@@ -1,44 +0,0 @@
-"""Example script for object tracking using DNT library."""
-
-import sys
-from pathlib import Path
-
-# Allow running this script directly from the repository root.
-ROOT = Path(__file__).resolve().parents[1]
-SRC = ROOT / "src"
-if str(SRC) not in sys.path:
-    sys.path.insert(0, str(SRC))
-
-from dnt.detect import Detector, DetectorModel
-from dnt.label import Labeler, LabelMethod
-from dnt.label.labeler import Labeler, LabelMethod
-from dnt.track import BOTSORTConfig, ReIDWeights, Tracker, interpolate_tracks_rts
-
-input_video = "/mnt/d/videos/sample/short-off-ramp.mp4"
-det_file = "/mnt/d/videos/sample/dets/short-off-ramp_det.txt"
-track_file = "/mnt/d/videos/sample/tracks/short-off-ramp_track.txt"
-track_interp_file = "/mnt/d/videos/sample/tracks/short-off-ramp_track_interp.txt"
-label_file = "/mnt/d/videos/sample/labels/short-off-ramp_track.mp4"
-label_interp_file = "/mnt/d/videos/sample/labels/short-off-ramp_track_interp.mp4"
-label_dir = "/mnt/d/videos/sample/labels/track_clips"
-
-detector = Detector(model=DetectorModel.RTDETRx, device="auto")
-# detector.detect(input_video, det_file)
-
-cfg = BOTSORTConfig(ReIDWeights.CLIP_VEHICLEID)
-tracker = Tracker(cfg, device="auto")
-tracks = tracker.track(det_file, track_file, input_video)
-
-labeler = Labeler(LabelMethod.CHROME_SAFE)
-labeler.draw_tracks(track_file=track_file, input_video=input_video, output_video=label_file, fill=True, alpha=0.2)
-
-tracks_interp = interpolate_tracks_rts(track_file=track_file, output_file=track_interp_file, verbose=True)
-labeler.draw_tracks(
-    track_file=track_interp_file,
-    input_video=input_video,
-    output_video=label_interp_file,
-    fill=True,
-    alpha=0.2,
-)
-
-print("ok")

dnt-0.3.2.2/examples/example_shapes.py

@@ -1,26 +0,0 @@
-from dnt.label import Labeler
-
-video_file = "/mnt/d/videos/hfst/Standard_SCU7WH_2022-09-13_1600.004.mp4"
-ouput_file = "/mnt/d/videos/hfst/labels/shapes.mp4"
-
-points = [[256, 228],[408, 337], 
-          [334, 212],[515, 297],
-          [403, 197],[596, 268],
-          [455, 187],[651, 247],
-          [507, 174],[693, 231]]
-
-lines = [[256, 228, 408, 337], 
-          [334, 212, 515, 297],
-          [403, 197, 596, 268],
-          [455, 187, 651, 247],
-          [507, 174, 693, 231]]
-
-polygons = [[[256, 228], [408, 337], [515, 297], [334, 212]],
-          [[334, 212], [515, 297], [596, 268], [403, 197]],
-          [[403, 197], [596, 268], [651, 247], [455, 187]],
-          [[455, 187], [651, 247], [693, 231], [507, 174]]]
-
-labeler = Labeler()
-labeler.draw_shapes(input_video=video_file, output_video=ouput_file, points = points, polygons=polygons, 
-                    point_color=(0,0,255), polygon_color=(0, 255, 0))
-

dnt-0.3.2.2/examples/example_shapes_v2.py

@@ -1,34 +0,0 @@
-import pandas as pd
-
-from dnt.label.labeler import Labeler
-
-video_file = "/mnt/d/videos/hfst/Standard_SCU7WH_2022-09-13_1600.004.mp4"
-ouput_file = "/mnt/d/videos/hfst/labels/shapes_v2.mp4"
-
-points = [
-    [256, 228],
-    [408, 337],
-    [334, 212],
-    [515, 297],
-    [403, 197],
-    [596, 268],
-    [455, 187],
-    [651, 247],
-    [507, 174],
-    [693, 231],
-]
-
-lines = [[256, 228, 408, 337], [334, 212, 515, 297], [403, 197, 596, 268], [455, 187, 651, 247], [507, 174, 693, 231]]
-
-polygons = [[25, 70], [25, 160], [110, 200], [200, 160], [200, 70], [110, 20]]
-
-polygons2 = [(25, 70), (25, 160), (110, 200), (200, 160), (200, 70), (110, 20)]
-
-results = []
-for i in range(1000):
-    results.append([i, "polylines", polygons2, (0, 255, 0), 1, 2, ""])
-
-df = pd.DataFrame(results, columns=["frame", "type", "coords", "color", "size", "thick", "desc"])
-
-labeler = Labeler()
-labeler.draw(video_file, output_video=ouput_file, draws=df)

dnt-0.3.2.2/examples/example_stop.py

@@ -1,50 +0,0 @@
-from dnt.analysis.stop import StopAnalyzer
-from dnt.label import Labeler
-from dnt.detect import Detector
-import pandas as pd
-import datetime    
-
-v_coords = [[(256, 228),(408, 337)], 
-          [(334, 212),(515, 297)],
-          [(403, 197),(596, 268)],
-          [(455, 187),(651, 247)],
-          [(507, 174),(693, 231)]]
-
-h_coords = [[(256, 228),(506, 175)], 
-          [(290, 256),(557, 188)],
-          [(345, 291),(635, 211)],
-          [(380, 313),(672, 225)],
-          [(408, 337),(691, 232)]]
-    
-event_dict = {
-        0: [0, 1, 2],    # Stop
-        1: [3]         # Pass
-    }
-
-track_file = "/mnt/d/videos/hfst/tracks/Standard_SCU7WH_2022-09-16_0630.02.001_track.txt"
-label_file = "/mnt/d/videos/hfst/tracks/Standard_SCU7WH_2022-09-16_0630.02.001_label.txt"
-result_file = "/mnt/d/videos/hfst/tracks/Standard_SCU7WH_2022-09-16_0630.02.001_result.txt"
-result_file2 = "/mnt/d/videos/hfst/tracks/Standard_SCU7WH_2022-09-16_0630.02.001_result2.txt"
-label_video = "/mnt/d/videos/hfst/labels/Standard_SCU7WH_2022-09-16_0630.02.001_label.mp4"
-video_file = "/mnt/d/videos/hfst/labels/Standard_SCU7WH_2022-09-16_0630.02.001_lines_3.mp4"
-
-analyzer = StopAnalyzer(h_coords=h_coords, v_coords=v_coords, event_dict=event_dict)
-analyzer.analysis(track_file=track_file, result_file=result_file, video_index=1, video_tot=1)
-
-results = pd.read_csv(result_file)
-results = results.loc[results['LANE']!=-1]
-results.to_csv(result_file, index=False)
-
-fps = Detector.get_fps(video_file)
-init_time = datetime.datetime.strptime('2022-09-16 10:47:08 AM', '%Y-%m-%d %I:%M:%S %p')
-StopAnalyzer.add_timestamp(result_file, result_file2, init_time=init_time, fps=fps)
-
-results = pd.read_csv(result_file2)
-results = results.loc[results['DURATION']>=1]
-results.to_csv(result_file2, index=False)
-
-StopAnalyzer.export_label(track_file, result_file2, label_file, label_field=6, event_label=['STOP', 'PASS'], vid_disp=False)
-
-labeler = Labeler(compress_message=False, label_fields=[6], nodraw_empty=True, zoom_factor=0.5)
-labeler.draw(label_file, video_file, label_video)
-

dnt-0.3.2.2/examples/example_track.py

@@ -1,45 +0,0 @@
-"""Example script for object tracking using DNT library."""
-
-import sys
-import time
-from pathlib import Path
-
-# Allow running this script directly from the repository root.
-ROOT = Path(__file__).resolve().parents[1]
-SRC = ROOT / "src"
-if str(SRC) not in sys.path:
-    sys.path.insert(0, str(SRC))
-
-from dnt.label.labeler import Labeler
-from dnt.track import BOTSORTConfig, Tracker
-
-"""
-input_video = "/mnt/d/videos/samples/ped_veh.mp4"
-det_file = "/mnt/d/videos/samples/dets/ped_veh_iou.txt"
-track_file = "/mnt/d/videos/samples/tracks/ped_veh_track.txt"
-label_video = "/mnt/d/videos/samples/labels/ped_veh_track.mp4"
-
-input_video = "/mnt/d/videos/samples/traffic.mp4"
-det_file = "/mnt/d/videos/samples/dets/traffic_iou.txt"
-track_file = "/mnt/d/videos/samples/tracks/traffic_track_dsort.txt"
-label_video = "/mnt/d/videos/samples/labels/traffic_track.mp4"
-"""
-input_video = "/mnt/d/videos/sample/traffic.mp4"
-det_file = "/mnt/d/videos/sample/dets/traffic_det.txt"
-track_file = "/mnt/d/videos/sample/tracks/traffic_track.txt"
-label_file = "/mnt/d/videos/sample/labels/traffic_track.mp4"
-
-tic = time.time()
-cfg = BOTSORTConfig()
-
-print(cfg)
-tracker = Tracker(cfg, device="auto")
-tracker.track(det_file, track_file, input_video)
-
-toc = time.time()
-print("Time:", int(toc - tic))
-
-labeler = Labeler()
-# labeler.draw_tracks(track_file=track_file, input_video=input_video, output_video=label_file, tail=100)
-
-print("ok")