import numpy as np
from numpy import random
# Generating an image of values between 1 and 255.
im_thresh = random.randint(1,256, (64,64))
# Set anything less than 255 to 0. Unnecessary if cv2 does this during threshold.
# Must go before the operation below in order not to set all values to 0.
im_thresh[im_thresh<255] = 0
# Set all values at indices where the array equals 255 to 1.
im_thresh[im_thresh==255] = 1