generate_defect.py 文件源码

def generate_defect_img(img,min_num,max_num,label_img):

    # label_img = np.zeros_like(img)

    # if random.random > 0.9:
    #     generate_crack(img,label_img,1,(0.01,0.05),6,(0.1,0.8))

    #method_list = [blur,scratch,spot]
    method_list = [blur,scratch,spot]


    num = random.randint(min_num,max_num)
    print num


    for i in range(num):
        fun_index = random.randint(0,len(method_list)-1)

        method_list[fun_index](img,label_img)

        # generate_blur(img,1,(0.05,0.3),(0.05,0.3))

        # generate_scratch(img,1,(0.001,0.05),20,(0.01,0.4))
        # generate_spot(img,1,(0.001,0.008),1.5)
    #return label_img