import os  
import cv2 
 
# 遍歷指定目錄，顯示目錄下的所有文件名
def CropImage4File(filepath,destpath):
  pathDir = os.listdir(filepath)  # 列出文件路徑中的所有路徑或文件
  for allDir in pathDir:
    child = os.path.join(filepath, allDir)
    dest = os.path.join(destpath,allDir)
    if os.path.isfile(child):
     image = cv2.imread(child) 
      sp = image.shape      #獲取圖像形狀：返回【行數(shù)值，列數(shù)值】列表
      sz1 = sp[0]         #圖像的高度（行 范圍）
      sz2 = sp[1]         #圖像的寬度（列 范圍）
      #sz3 = sp[2]        #像素值由【RGB】三原色組成
      
      #你想對(duì)文件的操作
      a=int(sz1/2-64) # x start
      b=int(sz1/2+64) # x end
      c=int(sz2/2-64) # y start
      d=int(sz2/2+64) # y end
      cropImg = image[a:b,c:d]  #裁剪圖像
      cv2.imwrite(dest,cropImg) #寫入圖像路徑
      
if __name__ == '__main__':
  filepath ='F:\\\maomi'       #源圖像
  destpath='F:\\maomi_resize'    # resized images saved here
  CropImage4File(filepath,destpath)

"""
處理數(shù)據(jù)集 和 標(biāo)簽數(shù)據(jù)集的代碼：（主要是對(duì)原始數(shù)據(jù)集裁剪）
  處理方式：分別處理
  注意修改 輸入 輸出目錄 和 生成的文件名
  output_dir = "./label_temp"
  input_dir = "./label"
"""
import cv2
import os
import sys
import time


def get_img(input_dir):
  img_paths = []
  for (path,dirname,filenames) in os.walk(input_dir):
    for filename in filenames:
      img_paths.append(path+'/'+filename)
  print("img_paths:",img_paths)
  return img_paths


def cut_img(img_paths,output_dir):
  scale = len(img_paths)
  for i,img_path in enumerate(img_paths):
    a = "#"* int(i/1000)
    b = "."*(int(scale/1000)-int(i/1000))
    c = (i/scale)*100
    time.sleep(0.2)
    print('正在處理圖像： %s' % img_path.split('/')[-1])
    img = cv2.imread(img_path)
    weight = img.shape[1]
    if weight>1600:             # 正常發(fā)票
      cropImg = img[50:200, 700:1500]  # 裁剪【y1,y2：x1,x2】
      #cropImg = cv2.resize(cropImg, None, fx=0.5, fy=0.5,
                 #interpolation=cv2.INTER_CUBIC) #縮小圖像
      cv2.imwrite(output_dir + '/' + img_path.split('/')[-1], cropImg)
    else:                    # 卷簾發(fā)票
      cropImg_01 = img[30:150, 50:600]
      cv2.imwrite(output_dir + '/'+img_path.split('/')[-1], cropImg_01)
    print('{:^3.3f}%[{}>>{}]'.format(c,a,b))

if __name__ == '__main__':
  output_dir = "../img_cut"      # 保存截取的圖像目錄
  input_dir = "../img"        # 讀取圖片目錄表
  img_paths = get_img(input_dir)
  print('圖片獲取完成 。。。！')
  cut_img(img_paths,output_dir)

#coding: utf-8
"""
采用多進(jìn)程加快處理。添加了在讀取圖片時(shí)捕獲異常，OpenCV對(duì)大分辨率或者tif格式圖片支持不好
處理數(shù)據(jù)集 和 標(biāo)簽數(shù)據(jù)集的代碼：（主要是對(duì)原始數(shù)據(jù)集裁剪）
  處理方式：分別處理
  注意修改 輸入 輸出目錄 和 生成的文件名
  output_dir = "./label_temp"
  input_dir = "./label"
"""
import multiprocessing
import cv2
import os
import time


def get_img(input_dir):
  img_paths = []
  for (path,dirname,filenames) in os.walk(input_dir):
    for filename in filenames:
      img_paths.append(path+'/'+filename)
  print("img_paths:",img_paths)
  return img_paths


def cut_img(img_paths,output_dir):
  imread_failed = []
  try:
    img = cv2.imread(img_paths)
    height, weight = img.shape[:2]
    if (1.0 * height / weight) < 1.3:    # 正常發(fā)票
      cropImg = img[50:200, 700:1500]   # 裁剪【y1,y2：x1,x2】
      cv2.imwrite(output_dir + '/' + img_paths.split('/')[-1], cropImg)
    else:                  # 卷簾發(fā)票
      cropImg_01 = img[30:150, 50:600]
      cv2.imwrite(output_dir + '/' + img_paths.split('/')[-1], cropImg_01)
  except:
    imread_failed.append(img_paths)
  return imread_failed


def main(input_dir,output_dir):
  img_paths = get_img(input_dir)
  scale = len(img_paths)

  results = []
  pool = multiprocessing.Pool(processes = 4)
  for i,img_path in enumerate(img_paths):
    a = "#"* int(i/10)
    b = "."*(int(scale/10)-int(i/10))
    c = (i/scale)*100
    results.append(pool.apply_async(cut_img, (img_path,output_dir )))
    print('{:^3.3f}%[{}>>{}]'.format(c, a, b)) # 進(jìn)度條（可用tqdm）
  pool.close()            # 調(diào)用join之前，先調(diào)用close函數(shù)，否則會(huì)出錯(cuò)。
  pool.join()             # join函數(shù)等待所有子進(jìn)程結(jié)束
  for result in results:
    print('image read failed!:', result.get())
  print ("All done.")



if __name__ == "__main__":
  input_dir = "D:/image_person"    # 讀取圖片目錄表
  output_dir = "D:/image_person_02"  # 保存截取的圖像目錄
  main(input_dir, output_dir)