#输入文件夹是input
输出文件夹是output
对input中图片提取边缘,存储在output中
import os
import argparse
import cv2
import sys
import numpy as np
parser = argparse.ArgumentParser(description=‘loading eval params’)
parser.add_argument(’–input_folder’, type=str, default=’./input’, help=‘path to the folder with test images’)
parser.add_argument(’–output_folder’, type=str, default=’./output’, help=‘path to the output folder’)
args = parser.parse_args()
if name == ‘main’:
args = parser.parse_args()
for filename in os.listdir(args.input_folder):
if filename.endswith(".png") or filename.endswith(".jpg"):
print(args.input_folder + ‘/’ + filename)
img = cv2.imread(args.input_folder + ‘/’ + filename)
gaussian_img = cv2.GaussianBlur(img, (3, 3), 0)
result = cv2.Canny(gaussian_img, 200, 480)
#filename = ‘%s/%s.jpg’ % (args.output_folder, ‘seg_’ + filename)
#cv2.imwrite(filename,result)
cv2.imwrite(args.output_folder + ‘/’ + filename, result)