[英]How to extract patches from an image in pytorch/tensorflow into 4 equal parts?
I am using a 16x16 colour image;我使用的是 16x16 彩色图像; I wrote small code for that but could not perform it precisely.
我为此编写了小代码,但无法精确执行。
import numpy as np
from patchify import patchify
image = cv2.imread('subbu_i.jpg')
print(image.shape)
patches = patchify(image, (4,4), step=1)
print(patches.shape)
Please help.请帮忙。
For Tensorflow, try tf.image.extract_patches对于 Tensorflow,尝试tf.image.extract_patches
import tensorflow as tf
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
image = Image.open('/content/image.png')
plt.imshow(image)
image = tf.expand_dims(np.array(image), 0)
image = tf.expand_dims(np.array(image), -1)
patches = tf.image.extract_patches(images=image,
sizes=[1, 4, 4, 1],
strides=[1, 4, 4, 1],
rates=[1, 1, 1, 1],
padding='VALID')
axes=[]
fig=plt.figure()
for i in range(4):
axes.append( fig.add_subplot(2, 2, i + 1) )
subplot_title=("Patch "+str(i + 1))
axes[-1].set_title(subplot_title)
patch = tf.reshape(patches[0, i, i], (4, 4))
plt.imshow(patch)
fig.tight_layout()
plt.show()
import tensorflow as tf
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
image = Image.open('/content/drive/MyDrive/subbu_i16.jpg')
plt.imshow(image)
image = tf.expand_dims(np.array(image),0) # To create the batch information
patches = tf.image.extract_patches(images=image,
sizes=[1, 4, 4, 1],
strides=[1, 4, 4, 1],
rates=[1, 1, 1, 1],
padding='VALID')
plt.figure(figsize=(10, 10))
for imgs in patches:
count = 0
for r in range(2):
for c in range(2):
ax = plt.subplot(2, 2, count+1)
plt.imshow(tf.reshape(imgs[r,c],shape=(4,4,3)).numpy().astype("uint8"))
count += 1
sample image result images Thank you @alonetogether示例图像结果图像谢谢@alonetogether
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.