OTVision 0.5.3__py3-none-any.whl

OTVision/view/view_transform.py

@@ -0,0 +1,140 @@
+"""
+OTVision gui module for transform.py
+"""
+
+# Copyright (C) 2022 OpenTrafficCam Contributors
+# <https://github.com/OpenTrafficCam
+# <team@opentrafficcam.org>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+
+import logging
+import shutil
+import tkinter as tk
+from pathlib import Path
+from tkinter import filedialog
+
+from OTVision.config import CONFIG, PAD
+from OTVision.helpers.files import get_files, replace_filetype
+from OTVision.helpers.log import LOGGER_NAME
+from OTVision.transform.reference_points_picker import ReferencePointsPicker
+from OTVision.transform.transform import main as transform
+from OTVision.transform.transform import write_refpts
+from OTVision.view.view_helpers import FrameRun
+
+log = logging.getLogger(LOGGER_NAME)
+
+
+class FrameTransform(tk.LabelFrame):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.frame_options = FrameTransformOptions(master=self)
+        self.frame_options.pack(**PAD, fill="x", expand=1, anchor="n")
+        self.frame_run = FrameRunTransformation(master=self)
+        self.frame_run.pack(**PAD, side="left", fill="both", expand=1, anchor="s")
+
+
+class FrameTransformOptions(tk.Frame):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        # Reference points
+        self.button_choose_refpts = tk.Button(
+            master=self, text="Choose reference points for selected", state=tk.DISABLED
+        )
+        self.button_choose_refpts.grid(row=0, column=0, columnspan=2, sticky="ew")
+        self.button_choose_refpts.bind("<ButtonRelease-1>", self.choose_refpts)
+        self.button_click_refpts = tk.Button(
+            master=self, text="New reference points", state=tk.DISABLED
+        )
+        self.button_click_refpts.grid(row=1, column=0, columnspan=2, sticky="ew")
+        self.button_click_refpts.bind("<ButtonRelease-1>", self.click_refpts)
+
+        # Overwrite
+        self.checkbutton_overwrite_var = tk.BooleanVar()
+        self.checkbutton_overwrite = tk.Checkbutton(
+            master=self,
+            text="Overwrite existing tracks",
+            variable=self.checkbutton_overwrite_var,
+        )
+        self.checkbutton_overwrite.grid(row=4, column=0, columnspan=2, sticky="w")
+        if CONFIG["TRANSFORM"]["OVERWRITE"]:
+            self.checkbutton_overwrite.select()
+
+    def choose_refpts(self, event):  # sourcery skip: use-named-expression
+        # Get selected files from files frame
+        selected_files = self.master.master.frame_files.get_selected_files()
+
+        if selected_files:
+            log.debug("choose refpts file for selected files")
+
+            # Show filedialog
+            refpts_file = filedialog.askopenfilename(
+                title="Select a reference points files",
+                filetypes=[(".otrfpts", ".otrfpts")],
+                initialdir=Path(selected_files[0]).parent,
+            )
+
+            # Check paths
+            refpts_file = get_files(paths=[refpts_file])[0]
+            log.debug(refpts_file)
+
+            # Copy refpts file for all selected
+            for selected_file in selected_files:
+                new_refpts_file = Path(selected_file).with_suffix(".otrfpts")
+                try:
+                    shutil.copy2(refpts_file, new_refpts_file)
+                except shutil.SameFileError:
+                    continue
+
+            # Update dict and treeview in files frame
+            self.master.master.frame_files.update_files_dict()
+
+    def click_refpts(self, event):
+        if selected_files := self.master.master.frame_files.get_selected_files():
+            log.debug("click and save refpts for selected files")
+
+            if refpts := ReferencePointsPicker(
+                video_file=Path(selected_files[0])
+            ).refpts:
+                # Save refpts for all selected files
+                for selected_file in selected_files:
+                    new_refpts_file = Path(selected_file).with_suffix(".otrfpts")
+                    write_refpts(refpts=refpts, refpts_file=new_refpts_file)
+
+                # Update dict and treeview in files frame
+                self.master.master.frame_files.update_files_dict()
+        else:
+            log.debug("No files selected")
+
+
+class FrameRunTransformation(FrameRun):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.button_run.bind("<ButtonRelease-1>", self.run)
+        if CONFIG["TRANSFORM"]["RUN_CHAINED"]:
+            self.checkbutton_run_chained.select()
+
+    def run(self, event):
+        tracks_files = replace_filetype(
+            files=self.master.master.frame_files.get_tree_files(),
+            new_filetype=CONFIG["DEFAULT_FILETYPE"]["TRACK"],
+        )
+        tracks_files = get_files(
+            paths=tracks_files,
+            filetypes=[CONFIG["DEFAULT_FILETYPE"]["TRACK"]],
+        )
+        log.info("Call transform from GUI")
+        transform(paths=tracks_files)

otvision-0.5.3.dist-info/METADATA

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: OTVision
+Version: 0.5.3
+Summary: OTVision is a core module of the OpenTrafficCam framework to perform object detection and tracking.
+Project-URL: Homepage, https://opentrafficcam.org/
+Project-URL: Documentation, https://opentrafficcam.org/overview/
+Project-URL: Repository, https://github.com/OpenTrafficCam/OTVision
+Project-URL: Issues, https://github.com/OpenTrafficCam/OTVision/issues
+Project-URL: Changelog, https://github.com/OpenTrafficCam/OTVision/releases
+Author-email: OpenTrafficCam contributors <team@opentrafficcam.org>, platomo GmbH <info@platomo.de>
+License-Expression: GPL-3.0-only
+License-File: LICENSE
+Keywords: OpenTrafficCam,Traffic Analysis,Traffic Counting,Trajectories
+Classifier: Development Status :: 4 - Beta
+Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.11
+Requires-Dist: av==13.0.0
+Requires-Dist: fire==0.7.0
+Requires-Dist: geopandas==1.0.1
+Requires-Dist: ijson==3.3.0
+Requires-Dist: moviepy==1.0.3
+Requires-Dist: numpy==1.26.4
+Requires-Dist: opencv-python==4.9.0.80
+Requires-Dist: pandas==2.2.2
+Requires-Dist: pyyaml==6.0.2
+Requires-Dist: setuptools==74.0.0
+Requires-Dist: torch==2.3.1
+Requires-Dist: torchvision==0.18.1
+Requires-Dist: tqdm==4.66.5
+Requires-Dist: ujson==5.10.0
+Requires-Dist: ultralytics==8.3.74
+Description-Content-Type: text/markdown
+
+# OTVision
+
+OTVision is a core module of the [OpenTrafficCam framework](https://github.com/OpenTrafficCam) to detect and track objects (road users) in videos recorded by [OTCamera](https://github.com/OpenTrafficCam/OTCamera) or other camera systems. On the resulting trajectories, one can perform traffic analysis using [OTAnalytics](https://github.com/OpenTrafficCam/OTAnalytics).
+
+Check out the [documentation](https://opentrafficcam.org/OTVision/) for detailed instructions on how to install and use OTVision.
+
+We appreciate your support in the form of both code and comments. First, please have a look at the [contribute](https://opentrafficcam.org/contribute) section of the OpenTrafficCam documentation.
+
+
+## License
+
+This software is licensed under the [GPL-3.0 License](LICENSE)

otvision-0.5.3.dist-info/RECORD

@@ -0,0 +1,50 @@
+OTVision/__init__.py,sha256=b2goS-TYrN9y8WHmjG22z1oVwCrjV0WtGJE32NJfvJM,1042
+OTVision/config.py,sha256=qCCg2MT_exTiTqEXq3pdqjhlfzQCg4yKKcGJSr7Wkp0,21951
+OTVision/dataformat.py,sha256=atssfeVcwXBjysXbVsgMbmtdMiChWzu8AwW4hK4GZw4,1884
+OTVision/version.py,sha256=rRi91utnulgYKaZW-qxdRDD7ufdJJ0G1IesKOtXP-NE,175
+OTVision/application/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/application/configure_logger.py,sha256=RmRgYkvSXWqpFX3N2n0h2LDna3uL8AY-yfVZU66t5q4,611
+OTVision/application/get_config.py,sha256=fRJZUWJvlTed6qriw82-B-05CAqfBma6NQ23WrKqYdE,771
+OTVision/application/detect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/application/detect/get_detect_cli_args.py,sha256=gezr17im8SwbuXW1suCodWRrFs8lSljNKu76SbWBgkY,265
+OTVision/application/detect/update_detect_config_with_cli_args.py,sha256=PxKosSR_4tlgJ-WpixrYHJJjUITpHGcD8ct9Fggpcys,3550
+OTVision/convert/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/convert/convert.py,sha256=vRIFcrAUf9MUWAuvF_q7JQ9aN89B1ouBOCtXrCLuDp8,11013
+OTVision/detect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/detect/builder.py,sha256=opd3RQ8ycm9Z75Z4OUiDXfP02bHvDslJcsfeKmLDwkw,1615
+OTVision/detect/cli.py,sha256=HpvdaXl_IJZe0CxZ9KmGzYDLL-zFABmfSV8aG3Rtamo,5454
+OTVision/detect/detect.py,sha256=O8xo0yH9yCmQKkvVqxkoJxQ9FcRPyf3s6IesgdEXcdo,10968
+OTVision/detect/otdet.py,sha256=rIiZR1J0_xdTVc6fwu8n2CrsIwpuD4TQiI6o99RTjog,3456
+OTVision/detect/yolo.py,sha256=cxRUxIchy_KYxkYUHEYyBCM4PB7d6E_nUDk0fja6XJY,9181
+OTVision/detect/plugin_av/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/detect/plugin_av/rotate_frame.py,sha256=4wJqTYI2HRlfa4p2Ffap33vLmKIzE_EwFvQraEkQ4R8,1055
+OTVision/domain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/domain/cli.py,sha256=RbcgaqULJuzYGG36970EYAxKoiLdBbGn2rX-JHpaH40,1000
+OTVision/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/helpers/date.py,sha256=XK0fmXMombQ2CJ2_RXjk-bc-R8qqXRK2rDmCMBi0_G0,854
+OTVision/helpers/files.py,sha256=JxO0ijcybBgIijIuA-9fnNhy-No0FAZCaE4RKMW0ie0,18116
+OTVision/helpers/formats.py,sha256=YLo_QLA2nhVREyv5N-xNW20c4nIL7DIF40E1lrsAyLE,4365
+OTVision/helpers/log.py,sha256=s5EYBboiNrkIKQE7WkPxnC0NJFL_shH7L5p7AI9IGJE,4544
+OTVision/helpers/machine.py,sha256=8Bz_Eg7PS0IL4riOVeJcEIi5D9E8Ju8-JomTkW975p8,2166
+OTVision/helpers/video.py,sha256=ZXG1rhD9OVk95mnfYVPbBwLKrRturgSRCsaOsGJVT2U,1324
+OTVision/track/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/track/iou.py,sha256=zpb4TbB34wZjUorhMLqMy6WspfsK-TkYvD75L-_ODM8,9491
+OTVision/track/iou_util.py,sha256=mqylrarqiauSewinxOm9CSa6dgVSl8G9bpCv--2m02M,4465
+OTVision/track/preprocess.py,sha256=P7w6N5OwIzdoDNwP5QcHHxRXwy0_ITz5Nkpk_P34UVs,14120
+OTVision/track/track.py,sha256=fKDP6jQ2pDLwxJOiDUYqs9fD3W67fNgwO3l8zUTYOw4,15344
+OTVision/transform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/transform/get_homography.py,sha256=29waW61uzCCB7tlBAS2zck9sliAxZqnjjOa4jOOHHIc,5970
+OTVision/transform/reference_points_picker.py,sha256=LOmwzCaqbJnXVhaTSaZtkCzTMDamYjbTpY8I99pN0rg,16578
+OTVision/transform/transform.py,sha256=cv8HIX-uPYVQWLGHU-8OrkQjJ2cOh3GIZ13wqeI431w,12455
+OTVision/view/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+OTVision/view/view.py,sha256=sBFGfr-lf8lG1BTaQ3Iu46GpZGhaVhSrqg8p5XUPJ4I,3302
+OTVision/view/view_convert.py,sha256=NUtmrmAAFCYdQS1U2av_eYUScXugQYjhD3e3H0pro6Q,5157
+OTVision/view/view_detect.py,sha256=E1JP9w5iv44hYkOr86In5d6V3sgn7Ld85KaRcY0oFZU,5951
+OTVision/view/view_helpers.py,sha256=a5yV_6ZxO5bxsSymOmxdHqzOEv0VFq4wFBopVRGuVRo,16195
+OTVision/view/view_track.py,sha256=ioHenonuTJQn8yDzlIDV8Jc1yzWA0GHbC9CotYGtxHQ,5389
+OTVision/view/view_transform.py,sha256=HvRd8g8geKRy0OoiZUDn_oC3SJC5nuXhZf3uZelfGKg,5473
+OTVision/view/helpers/OTC.ico,sha256=G9kwlDtgBXmXO3yxW6Z-xVFV2q4nUGuz9E1VPHSu_I8,21662
+otvision-0.5.3.dist-info/METADATA,sha256=y9bE7wsK_cHxpU9EmFkdbuPdkaxB4PbbHgGvZTTE5AA,2244
+otvision-0.5.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+otvision-0.5.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+otvision-0.5.3.dist-info/RECORD,,

otvision-0.5.3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any