Report an error： RuntimeWarning: invalid value encountered in cast aff_img = Image.fromarray((affinity_map * 255).astype(np.uint8))

Report an error： RuntimeWarning: invalid value encountered in cast aff_img = Image.fromarray((affinity_map * 255).astype(np.uint8)) #366