Update main.py

main.py

@@ -509,58 +509,29 @@ class HFVideoDataSet(VideoDataSet):
                     shutil.rmtree(self.temp_dir)
                 raise e
         
-        # CRITICAL FIX: Ensure the temp directory path ends with a slash
-        # This is needed because the parent VideoDataSet class likely does string concatenation
-        if not self.temp_dir.endswith(os.sep):
-            self.temp_dir = self.temp_dir + os.sep
-        
+        # Set the feature path to our temp directory
         opt['video_feature_all_test'] = self.temp_dir
         print(f"Set video_feature_all_test to: {opt['video_feature_all_test']}")
         
-        # Store reference for potential path fixes
-        self.video_name_for_single_video = video_name
-        
         # Initialize parent class
         try:
             super().__init__(opt, subset, video_name)
             print(f"Successfully initialized dataset with {len(self.video_list)} videos")
         except Exception as e:
             print(f"Error initializing parent VideoDataSet: {str(e)}")
-            # Debug: Let's see what files are actually in the temp directory
-            temp_dir_clean = self.temp_dir.rstrip(os.sep)
-            if os.path.exists(temp_dir_clean):
-                files_in_temp = os.listdir(temp_dir_clean)
-                print(f"Files in temp directory: {files_in_temp}")
-                
-                # Try to find the expected file with different possible constructions
-                expected_file_1 = os.path.join(temp_dir_clean, f"{video_name}.npz")
-                expected_file_2 = f"{temp_dir_clean}{video_name}.npz"
-                expected_file_3 = f"{self.temp_dir}{video_name}.npz"
-                
-                print(f"Expected file path 1: {expected_file_1} - Exists: {os.path.exists(expected_file_1)}")
-                print(f"Expected file path 2: {expected_file_2} - Exists: {os.path.exists(expected_file_2)}")
-                print(f"Expected file path 3: {expected_file_3} - Exists: {os.path.exists(expected_file_3)}")
-            
             # Clean up temp directory on error
-            if hasattr(self, 'temp_dir') and os.path.exists(temp_dir_clean):
-                shutil.rmtree(temp_dir_clean)
+            if hasattr(self, 'temp_dir') and os.path.exists(self.temp_dir):
+                shutil.rmtree(self.temp_dir)
             raise e
     
     def __del__(self):
         # Clean up temporary directory
         try:
-            if hasattr(self, 'temp_dir'):
-                temp_dir_clean = self.temp_dir.rstrip(os.sep)
-                if os.path.exists(temp_dir_clean):
-                    shutil.rmtree(temp_dir_clean)
-                    print(f"Cleaned up temp directory: {temp_dir_clean}")
+            if hasattr(self, 'temp_dir') and os.path.exists(self.temp_dir):
+                shutil.rmtree(self.temp_dir)
+                print(f"Cleaned up temp directory: {self.temp_dir}")
         except Exception as e:
             print(f"Warning: Could not clean up temp directory: {e}")
-    
-    def get_video_feature_path(self, video_name):
-        """Override to ensure proper path construction"""
-        temp_dir_clean = self.temp_dir.rstrip(os.sep)
-        return os.path.join(temp_dir_clean, f"{video_name}.npz")
 
 def eval_frame(opt, model, dataset):
     """Evaluate model frame by frame"""