opencv dct变换报错

进行DCT变换的时候总是报错
OpenCV Error: Assertion failed (type == CV_32FC1 || type == CV_64FC1) in cv::dct, file …\modules\core\src\dxt.cpp, line 3603
(117, 128, 3)
Traceback (most recent call last):
File “dct.py”, line 15, in
img_dct = cv2.dct(img1) #\u8fdb\u884c\u79bb\u6563\u4f59\u5f26\u53d8\u6362
cv2.error: …\modules\core\src\dxt.cpp:3603: error: (-215) type == CV_32FC1 || type == CV_64FC1 in function cv::dct

后面发现是需要对图片取一个通道，并转换成float格式进行dct才能成功变换

import cv2
import numpy as np
img = cv2.imread(r'F:/SRnet/train_data/stego/000421.jpg')

# cv2.imshow('image', img)
# cv2.waitKey(0)
img = img[:, :, 0]
img = np.float32(img)
img = cv2.dct(img)
img = np.array(img)
print(img.shape)