updated main

malikaratnayake · Aug 15, 2024 · 9f7ca5d · 9f7ca5d
1 parent d49d99f
commit 9f7ca5d
Show file tree

Hide file tree

Showing 10 changed files with 7 additions and 6 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/.gitignore b/.gitignore
@@ -14,3 +14,4 @@ polytrack/__pycache__/config.cpython-310.pyc
 polytrack/__pycache__/flowers.cpython-310.pyc
 polytrack/__pycache__/general.cpython-310.pyc
 polytrack/__pycache__/tracker.cpython-310.pyc
+data/Ratnayake2023_processed/
diff --git a/PolyTrack.py b/PolyTrack.py
@@ -21,7 +21,6 @@
 def main(_argv):
     start_time = datetime.now()
     start_time_py = time.time()
-    print("Start:  " + str(start_time))
     nframe = 0
     total_frames = 0
     predicted_position =[]
@@ -38,6 +37,7 @@ def main(_argv):
         print('===================' + str(video_name) + '===================')
 
         video = str(pt_cfg.POLYTRACK.INPUT_DIR) + str(video_name)
+        # TrackInsect = InsectTracker(video)
 
         try:
             vid = cv2.VideoCapture(int(video))

diff --git a/data/.DS_Store b/data/.DS_Store
diff --git a/data/output/.DS_Store b/data/output/.DS_Store
diff --git a/data/output/CV4A/video_details.txt b/data/output/CV4A/video_details.txt
diff --git a/polytrack/__pycache__/config.cpython-310.pyc b/polytrack/__pycache__/config.cpython-310.pyc
diff --git a/polytrack/__pycache__/tracker.cpython-310.pyc b/polytrack/__pycache__/tracker.cpython-310.pyc
diff --git a/polytrack/config.py b/polytrack/config.py
@@ -12,11 +12,11 @@
 __PT.POLYTRACK                 = edict()
 
 __PT.POLYTRACK.INPUT          = './data/video/cam_3_N_video_20210308_122000' # File location of the input video file (If only a single video needs to be processed)
-__PT.POLYTRACK.INPUT_DIR      = './data/video/bees-edge/unprocessed/' # Specify the directory of the input video files 
-__PT.POLYTRACK.COMPRESSED_VIDEO  = False # Set TRUE if the input video is motion compressed (beta).
+__PT.POLYTRACK.INPUT_DIR      = '/Users/mrat0010/Documents/GitHub/Polytrack_PUB/data/Ratnayake2023_processed/cam_1_N_video_20210315_132804.h264' # Specify the directory of the input video files 
+__PT.POLYTRACK.COMPRESSED_VIDEO  = True # Set TRUE if the input video is motion compressed (beta).
 __PT.POLYTRACK.VIDEO_EXT      = 'avi' # Video extension
-__PT.POLYTRACK.CONTINUOUS_VIDEO = True # For processing multiple video files - TRUE, if video files are sequencial
-__PT.POLYTRACK.OUTPUT         = './data/output/' # Output directory
+__PT.POLYTRACK.CONTINUOUS_VIDEO = False # For processing multiple video files - TRUE, if video files are sequencial
+__PT.POLYTRACK.OUTPUT         = './data/output/CV4A/' # Output directory
 __PT.POLYTRACK.VIDEO_START_TIME = None #Start time of the video. To be extracted from the video file name (default setting -> None)
 __PT.POLYTRACK.CURRENT_VIDEO_DETAILS = None #Video details. To be extracted from the video file name (default setting -> None)
 __PT.POLYTRACK.INSECT_COUNT = 0 

diff --git a/polytrack/tracker.py b/polytrack/tracker.py
@@ -221,7 +221,7 @@ def get_video_info(_video_info_filepath:str) -> tuple:
 
         try:
 
-            csv_file = str(_video_info_filepath) + 'video_info.csv'
+            csv_file = str(_video_info_filepath) + '_video_info.csv'
 
             with open(csv_file, "r", encoding="utf-8") as csv_file:
                 csv_reader = csv.reader(csv_file)