Update preprocessing.py

preprocessing.py

@@ -14,7 +14,7 @@ def get_video_file(file_path, ftp_password):
     ftp_port = 21
     ftp_user = "donghuna_ftp"
     # folder_path = "homes/donghuna/database/Diving48_rgb/rgb/"
-    folder_path = "web/donghuna.com/inference-endpoints/dive-sequence-classification/"
+    # folder_path = "web/donghuna.com/inference-endpoints/dive-sequence-classification/"
     
     # FTP 연결 설정
     ftp = FTP()
@@ -25,22 +25,22 @@ def get_video_file(file_path, ftp_password):
     local_path = "test.mp4"
     
     with open(local_path, 'wb') as local_file:
-        ftp.retrbinary(f'RETR {folder_path}{file_path}', local_file.write)
+        ftp.retrbinary(f'RETR {file_path}', local_file.write)
 
 
-def read_video(file_path, num_frames=24, target_size=(224, 224)):
+def read_video(file_path, ftp_password, num_frames=24, target_size=(224, 224)):
     get_video_file(file_path, ftp_password)
 
+    local_path = "test.mp4"
+    logging.info(f"Reading video from: {local_path}")
+    container = av.open(local_path)
+    frames = []
+    for frame in container.decode(video=0):
+        frames.append(frame.to_ndarray(format="rgb24").astype(np.uint8))
     
-    # logging.info(f"Reading video from: {file_path}")
-    # container = av.open(file_path)
-    # frames = []
-    # for frame in container.decode(video=0):
-    #     frames.append(frame.to_ndarray(format="rgb24").astype(np.uint8))
-    
-    # sampled_frames = sample_frames(frames, num_frames)
-    # processed_frames = preprocess_frames(sampled_frames, target_size)
-    # return processed_frames
+    sampled_frames = sample_frames(frames, num_frames)
+    processed_frames = preprocess_frames(sampled_frames, target_size)
+    return processed_frames
 
 def sample_frames(frames, num_frames):
     total_frames = len(frames)