浏览代码

fix expected extension

Jérôme BUISINE 3 年之前
父节点
当前提交
cbf0d1facd
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      run/extract_specific_png.py

+ 1 - 1
run/extract_specific_png.py

@@ -35,7 +35,7 @@ def main():
         # replace expected Samples value
         img_data = img.split('-')
         img_data[-2] = "S" + p_samples
-        img_data[-1] = output_index
+        img_data[-1] = output_index + ".png"
 
         output_path = '-'.join(img_data)