Folder Concept Captioning Script

이 스크립트는 DonMischo가 작성한 것으로, 저는 약간 수정하여 더 유연하게 만들었습니다. 여러 이미지 형식을 지원하도록 추가했으며, 이제 매번 수동으로 스크립트를 편집할 필요 없이 스크립트를 실행할 때 모델 이름을 입력받습니다.

스크립트를 한 폴더에 넣고, 각 개념별로 이미지를 하나의 폴더에 정리한 데이터셋을 준비하세요. 그런 다음 모든 개념 폴더를 스크립트와 동일한 폴더에 두세요. 다음과 같이 구조를 구성하세요:

그런 다음 스크립트를 실행하고 모델 이름을 입력하세요. 이 이름은 캡션에 접두사로 사용됩니다. 그러면 각 이미지에 대해 개념 폴더 내에 캡션 파일을 생성하며, 개념 폴더 이름을 캡션의 일부로 사용합니다.