#coding=utf-8
import cv2
import os

path='/home/fhy145/workspace/bm2/dataset/winter/bgr1/'
img_paths=os.listdir(path)
#print(img_paths)
# img_paths.sort(key=lambda x: int(x[:-4]))  # 倒着数第四位'.'为分界线，按照‘.'左边的数字从小到大排序
# print(img_paths)
path_obj='/home/fhy145/workspace/bm2/dataset/winter/bgr2/'
 
# 判断结果
if not os.path.exists(path_obj):
    # 如果不存在则创建目录
    # 创建目录操作函数
    os.makedirs(path_obj)
    print(path_obj," create over!")
else:
    print(path_obj," exist!")

i=0
for img in img_paths:
    i+=1
    abs_path = os.path.join(path,img)
    img_name=img.split('.')[0]
    print(img_name)
    img_mat = cv2.imread(abs_path)
    # cv2.imshow("img",img_mat)
    # cv2.waitKey(1000)
    cv2.imwrite(path_obj+img_name+".jpg",img_mat)
    print(i)