今天再记录一下代码Hashing_for_geo-localization的debug日常
feature_extraction/view_converter.py
for img in images: signal = imread(input_dir + img) image = sample_bilinear(np.array(signal), x, y) imsave(output_dir + img.replace('.jpg', '.png'), image)
一直会报错Cannot handle this data type: (1, 1, 3),<f8
修改之后可以跑通
# Assuming sample_bilinear returns a NumPy array def scale_and_convert_to_uint8(image): # Scale the values to the range [0, 255] scaled_image = ((image - image.min()) / (image.max() - image.min()) * 255).astype(np.uint8) return scaled_image for img in images: signal = imread(input_dir + img) image = sample_bilinear(np.array(signal), x, y) # Scale and convert to uint8 scaled_image = scale_and_convert_to_uint8(image) imsave(output_dir + img.replace('.jpg', '.png'), scaled_image)