import os
import random

# 이미지들이 있는 디렉토리 경로
images_dir = '/data/ltb/ultralytics/objects365_mmyolo_coco/images/'

# 디렉토리 내 파일을 모두 불러와 확장자가 jpg/jpeg/png인 것만 필터링
all_files = [f for f in os.listdir(images_dir)
             if f.lower().endswith(('.jpg', '.jpeg', '.png'))]

# 샘플 개수 설정 (파일 수가 10개 미만이면 전체를 다 가져옵니다)
sample_size = min(10, len(all_files))

# 무작위 추출
sample_files = random.sample(all_files, sample_size)

# 결과 출력
print(f"샘플 파일 {sample_size}개:")
for fname in sample_files:
    print(fname)