drbh
commited on
Commit
·
5ddd783
1
Parent(s):
63f348a
fix: adjust warning
Browse files- test_kernel.py +1 -1
test_kernel.py
CHANGED
|
@@ -25,5 +25,5 @@ gray_tensor = img2gray_lib.img2gray(img_tensor).squeeze()
|
|
| 25 |
print(gray_tensor.shape) # HW
|
| 26 |
|
| 27 |
# save the output image
|
| 28 |
-
gray_img = Image.fromarray(gray_tensor.cpu().numpy().astype(np.uint8)
|
| 29 |
gray_img.save("kernel-builder-logo-gray2.png")
|
|
|
|
| 25 |
print(gray_tensor.shape) # HW
|
| 26 |
|
| 27 |
# save the output image
|
| 28 |
+
gray_img = Image.fromarray(gray_tensor.cpu().numpy().astype(np.uint8))
|
| 29 |
gray_img.save("kernel-builder-logo-gray2.png")
|