CodeHoon

이미지 데이터셋을 다루거나 정제된 이미지 폴더의 파일 수를 파악해야 하는 경우가 많습니다.
특히 하위 폴더가 복잡하게 구성된 구조에서는 재귀적으로 전체 폴더를 순회하면서 이미지 파일을 세는 로직이 필요합니다.

Java에서는 이를 직접 재귀함수로 구현해야 하지만, Python은 os.walk() 라는 내장 모듈로 매우 간단하게 처리할 수 있습니다.

이번 글에서는 Python의 os.walk()를 이용해 폴더 내 이미지 파일 수를 집계하는 방법을 소개하고,
Java 방식과 비교해 생산성과 코드 간결성의 차이도 함께 알아보겠습니다.

1. Java에서의 폴더 탐색: 수동 재귀의 한계

Java에서 폴더 내의 파일을 탐색하기 위해서는 일반적으로 다음과 같은 수동 재귀 로직이 필요합니다.

public static int countImages(File dir) {
    int count = 0;
    File[] files = dir.listFiles();
    if (files == null) return 0;

    for (File file : files) {
        if (file.isDirectory()) {
            count += countImages(file); // 재귀 호출
        } else if (file.getName().matches("(?i).*\\.(png|jpg|jpeg)$")) {
            count++;
        }
    }
    return count;
}

이 방식은 다음과 같은 문제점이 있습니다:

• 예외 처리 및 디렉토리 접근 검증 코드를 별도로 작성해야 함
• 파일 확장자 필터링을 위한 정규식 작성 필요
• 코드 길이가 길고 가독성이 떨어짐

2. Python의 os.walk: 자동 재귀의 강력함

2.1 os.walk란?

os.walk(path)는 지정한 경로부터 시작해 모든 하위 디렉토리를 자동으로 순회하며, 각 폴더 내의 파일과 디렉토리 정보를 튜플 형태로 반환해주는 제너레이터 함수입니다.

for root, dirs, files in os.walk(target_path):
    # root: 현재 탐색 중인 폴더 경로
    # dirs: 현재 폴더 내의 하위 폴더 목록
    # files: 현재 폴더 내의 파일 목록

2.2 이미지 파일 필터링 예시

이미지 파일 확장자를 필터링하려면 files 리스트에 대해 다음과 같은 list comprehension을 사용하면 됩니다:

image_files = [
    f for f in files
    if f.lower().endswith((".png", ".jpg", ".jpeg"))
]

3. 실전 코드 예시 및 설명

다음은 Python 코드 전체입니다. 이 코드는 정제/ 폴더 하위에 있는 모든 이미지(.png, .jpg, .jpeg) 파일의 개수를 계산하고,
결과를 JSON 파일로 저장합니다.

FastAPI 를 사용하였습니다

@dashboard_router.post("/update-image-count", response_model=ImageCountResponse)
def update_image_count_json():
    try:
        JSON_PATH = os.path.join(ROOT_DIR, "etc/refinedDataCount.json")

        # JSON 로드 또는 초기화
        if os.path.exists(JSON_PATH):
            with open(JSON_PATH, "r", encoding="utf-8") as f:
                data_count = json.load(f)
        else:
            data_count = {}

        # 탐색할 루트 경로 설정
        refined_root = os.path.join(ROOT_DIR, "정제")

        image_count = 0

        # os.walk를 통한 자동 재귀 탐색
        for root, dirs, files in os.walk(refined_root):
            image_count += len([
                f for f in files
                if f.lower().endswith((".png", ".jpg", ".jpeg"))
            ])

        # 결과 저장
        data_count["total_png"] = image_count
        os.makedirs(os.path.dirname(JSON_PATH), exist_ok=True)
        with open(JSON_PATH, "w", encoding="utf-8") as f:
            json.dump(data_count, f, ensure_ascii=False, indent=4)

        print(f"✅ 총 이미지 개수: {image_count} → {JSON_PATH} 저장됨")

        return {"success": True, "message": "이미지 개수 업데이트 완료", "image_count": image_count}
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Error: {str(e)}")

4. 핵심 포인트 정리

Python은 반복적이고 재귀적인 작업을 매우 간단하게 처리할 수 있도록 다양한 내장 기능을 제공합니다. os.walk는 그 대표적인 예시로, 파일 시스템을 다루는 모든 작업에 강력한 도구가 됩니다.

5. 마무리

Python의 os.walk()는 재귀적으로 파일 시스템을 탐색할 수 있는 매우 강력하고도 간단한 도구입니다. Java와 같은 언어에서는 재귀 구조를 직접 설계해야 하는 반면, Python은 이러한 반복 작업을 몇 줄의 코드로 깔끔하게 처리할 수 있습니다.

이미지 파일 개수 집계, 대용량 폴더 분석, 정제 데이터 자동화 등에 활용하면 실무에서 효율적으로 사용할 수 있습니다.
지금 바로 여러분의 프로젝트에 적용해보세요!

📌 도움이 되셨다면 댓글이나 공감 부탁드립니다
궁금한 점이 있다면 언제든지 질문해주세요!

사용환경: WINDOW, LINUX, Python 3.10.2, PHP 8.0.3 (이전 버전도 가능)

지난 포스팅

PHP 에서 Python 에 파라미터를 전달하는 방법을 포스팅하겠습니다.

<?php
$data_value = '첫번째';
$data_value2 = '두번째';
$output = shell_exec("python-text.py".' '.$data_value.' '.$data_value2);
echo $output;

shell_exec 를 사용해주면 되는데, 주의할 점이 있습니다.

바로 띄어쓰기 입니다.

1. 띄어쓰기를 통해 매개변수를 구분한다.
2. 만약 데이터에 '안녕하세요  훈입니다.' 와 같이 띄어쓰기가 있는 데이터는 별도의 데이터로 출력됩니다.

# python

import sys
import io	#한글처리

sys.stdout = io.TextIOWrapper(sys.stdout.detach(), encoding='utf-8')		#한글처리
sys.stderr = io.TextIOWrapper(sys.stderr.detach(), encoding='utf-8')		#한글처리

for arg in sys.argv:
    print(arg)

sys.exit()

반복문을 통하여 전달된 인자 arg 를 출력해주면 됩니다. 

arg 의 결과물은 배열(Array)인데, 첫번째 인자출력물로 현재 경로를 출력합니다.  사용에 주의 바랍니다.
만약 값이 전달되지 않았을 때는, 경로만 출력이 되기 때문에 값이 잘 전달되었는지 확인이 가능합니다.

결과

띄어쓰기가 있는 데이터를 전달해주었을때

개행되어서 출력되는 것을 확인하실 수 있습니다.

사용환경: WINDOW, LINUX, Python 3.10.2, PHP 8.0.3 (이전 버전도 가능)

PHP에서 Python의 파일을 호출하는 방법

PHP에서 shell_exec 를 통하여 Python 파일을 호출할 수 있다.

// 사용할 php파일

$output = shell_exec("python-text.py");	   // python-text.py : 파이썬 파일 이름 
echo $output;

최대한 내용을 짧게 하고 싶어서 명령어 없이 작성하였다.

다른 방법도 있다,

// 사용할 php 파일

$output = exec( "python python-file-name.py" );
echo $output;

파이썬 파일 이름 앞에 py, python, python3 이라든지 파이썬 명령어를 사용해도 무방하다. 

// python.py

helloThere = 'hello'
print(helloThere)

결과 출력 - hello

다음 포스팅은 php에서 매개변수(파라미터)를 전달하는 방법을 포스팅하겠습니다.