generate_defect.py 文件源码

def main():
    root = './resize_4096'
    save_dir = './generate_defect1_again'



    dataset_generate(root,save_dir)

    # cv2.imshow("img", img)

    # cv2.waitKey(0)

# def main():
#     img = cv2.imread('./def_1.png',cv2.cv.CV_LOAD_IMAGE_GRAYSCALE)
#     print img.shape

#     cv2.namedWindow('label',cv2.WINDOW_NORMAL)
#     cv2.namedWindow('img',cv2.WINDOW_NORMAL)

#     # label = generate_defect_img(img,1,5)
#     label_img = np.zeros_like(img)

#     #generate_blur(img,label_img,1,(0.05,0.1),(0.05,0.1))

#     #generate_scratch(img,label_img,1,(0.0001,0.002),5,(0.005,0.4))

#     #generate_spot(img,label_img,1,(0.0001,0.0005),2)


#     cv2.imshow("label", label_img)
#     cv2.imshow("img", img)
#     cv2.waitKey(0)