pytorch rgb to grayscale code example

Example 1: opencv grayscale to rgb

backtorgb = cv2.cvtColor(gray,cv2.COLOR_GRAY2RGB)

Example 2: cvtcolor rgb to gray

cv::Mat img
cv::cvtColor(img, img, CV_BGR2GRAY);