디렉토리에서 이미지 시퀀스 찾기 - os.walk()보다 빠름?

Question

os.walk를 사용하지 않고 디렉토리를 통과하는 스크립트 (예 : os.walk와 거의 동일)를 작성하려고합니다. 그 이유는 각 디렉토리가 이미지 시퀀스로 가득 차고 os.walk가 너무 많은 시간을 들여야하기 때문입니다.

내 의도는 각 이미지 시퀀스의 파일 이름을 수집하는 것입니다. 그래서 현재 시퀀스의 단일 이미지가 발견 되 자마자 루프에서 벗어납니다. 디렉토리에 단 하나의 이미지 시퀀스가 ​​있으면 잘 작동하지만 디렉토리에 이미지 시퀀스가 ​​2 개 이상 있으면 어떻게됩니까? 이것은 제가 풀려나는 곳입니다.

def find_all_ImageSequences(dir): 

    for object in os.listdir(dir): 
     if os.path.isdir(dir + "\\" + str(object)): 
      find_all_ImageSequences(dir + "\\" + str(object)) 
     else: 
      if object.endswith(('.dpx','.jpg','.jpeg','.exr','.tif')): 
       Image_Sequences_List.append(dir + "\\" + str(object)) 
       image_filename = object.split(".")[0] 
       extras = [x for x in os.listdir(dir) if x.split(".")[0] != image_filename] 
       if extras: 
        "try to take the 1st image and add it to my Image_Sequences_List" 
       break 

은 "엑스트라"지능형리스트가 두 번째 이미지 시퀀스를 찾았다 후에도, 그 디렉토리에있는 모든 개체를 통해 보는 것 :

이것은 내가 지금까지있는 것입니다. 다음 이미지 시퀀스의 첫 번째 프레임이 발견되는 즉시이를 중단 할 수 있습니까?

내가 사용하는 방법에 대한 생각 :

if any([x for x in os.listdir(dir) if x.split(".")[0] != image_filename]): 
    "do something" 

그러나 이것은 내가 그 다음 순서를 찾기 위해 다시 현재 디렉토리를 반복 할 것 같은 카운터 생산성을 보였다.

이 작업을 수행하는 가장 좋은 방법은 무엇입니까? 성능을 향상 인 os.walk 교체가 https://github.com/benhoyt/scandir을 확인하려면

def find_all_ImageSequences(dir): 
    seen = set() 
    for object in os.listdir(dir): 
     if os.path.isdir(dir + "\\" + str(object)): 
      find_all_ImageSequences(dir + "\\" + str(object)) 
     else: 
      if object.endswith(('.dpx','.jpg','.jpeg','.exr','.tif')): 
       image_filename = object.split(".", 1)[0] 
       if image_filename not in seen: 
        Image_Sequences_List.append(dir + "\\" + str(object)) 
        seen.add(image_filename) 

---

:

많은 감사

Answer 1

약간의 시간이 지나면 이것을 약화시키는 것이 가장 빠른 해결책이라는 것을 알게되었습니다.

def find_all_ImageSequences(dir): 

    child_directories = [x for x in os.listdir(dir) if len(x.split(".")) == 1] 
    if len(child_directories) >= 1: 
     for directory in child_directories: 
      if os.path.isdir(dir + "\\" + directory): 
       find_all_ImageSequences(dir + "\\" + directory) 

    for object in os.listdir(dir): 
     if object.endswith(('.dpx','.jpg','.jpeg','.exr','.tif')): 
      image_filename = object.split(".")[0] 
      Image_Sequences_List.append(dir + "\\" + str(object)) 
      if any([x for x in os.listdir(dir) if x.split(".")[0] != image_filename and len(x.split(".")) > 1]): 
       extras = set([x.split(".")[0] for x in os.listdir(dir) if x.split(".")[0] != image_filename]) 
       for i in extras: 
        Image_Sequences_List.append(i) 
      break 

내가 표준 os.walk()이 기능을 비교하는 빠른 테스트했다 :

• os.walk()를 5 초 걸렸습니다() 평균
• find_all_ImageSequences 53 초 걸렸습니다 평균

다른 이미지 시퀀스가없는 경우이 속도는 주로 디렉토리를 벗어납니다. 도움을 준 Janne Karila에게 감사드립니다. set()을 사용하면 매우 유용합니다.

Answer 2

이 코드는 이미 추가 된 파일 이름을 추적하기 위해 세트를 사용 Windows에서 약 8-9 배, Linux 및 Mac OS X에서 약 2-3 배 빠르다 고 주장했습니다.