create_train.py error

[Dong]

find some error here

 a = 1 - 0.5 * uniform(0, 1)


        m = gt_image.shape[0]
        n = gt_image.shape[1]

        rep_atmosphere = np.tile(np.reshape(A, [1, 1, 3]), [m, n, 1])

    ...
        h5f.create_dataset('atom',data=rep_atmosphere)

change to:

        a = 1 - 0.5 * uniform(0, 1)


        m = gt_image.shape[0]
        n = gt_image.shape[1]

        rep_atmosphere = np.tile(np.tile(a, [1, 1, 3]), [m, n, 1])



    ...
        h5f.create_dataset('ato',data=rep_atmosphere)

[xieliangru]

i change it to:
a = 1 - 0.5 * uniform(0, 1)

    m = gt_image.shape[0]
    n = gt_image.shape[1]

    rep_atmosphere = np.tile(np.full([1, 1, 3],a), [m, n, 1])



...
    h5f.create_dataset('ato',data=rep_atmosphere)

[beijing1214]

thx very much. I got confused because the variable A