import os
import cv2

def process_directory(directory, output_file):
    for root, dirs, files in os.walk(directory):
        if not dirs:
            # 더 이상 하위 디렉토리가 없는 경우
            print("Directory:", root)
    
            # 이미지 파일 경로 검사
            image_found = False
            for filename in files:
                if filename in ["im1.png","im1.jpg"]:
                    image_found = True
                    image_path = os.path.join(root, filename)
                
                    # 이미지 읽기
                    image = cv2.imread(image_path)[:, :, ::-1]
                    if image is not None:
                        print("Image:", image_path)
    
                    # 이미지를 찾지 못한 경우 경로를 파일에 쓰기
                    if not image_found:
                        with open(output_file, "a") as f:
                            f.write(root + "\n")
    
if __name__ == "__main__":
    # 디렉토리 경로 지정
    start_directory = "../../ins4/Triplet_250p_new/sequences"
    
    # 결과를 저장할 텍스트 파일 경로 지정
    output_file = "missing_images.txt"
    
    # 텍스트 파일 초기화
    open(output_file, "w").close()
    
    # 시작 디렉토리에서 순회 시작
    process_directory(start_directory, output_file)