Update preprocessing.py

preprocessing.py

@@ -14,6 +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/"
     
     # FTP 연결 설정
     ftp = FTP()
@@ -21,22 +22,25 @@ def get_video_file(file_path, ftp_password):
     ftp.login(user=ftp_user, passwd=ftp_password)
     ftp.set_pasv(True)
 
-    video_path = "test.mp4"
+    local_path = "test.mp4"
     
-    with open(video_path, 'wb') as local_file:
-        ftp.retrbinary(f'RETR {file_path}', local_file.write)
+    with open(local_path, 'wb') as local_file:
+        ftp.retrbinary(f'RETR {folder_path}{file_path}', local_file.write)
 
 
 def read_video(file_path, num_frames=24, target_size=(224, 224)):
+    get_video_file(file_path, ftp_password)
+
+    
     # 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))
+    # 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)