I have three images to check. How do you know which one is on the test image?
import cv2 import numpy as np CLASS = [\'1.png\', \'2.png\', \'3.png\'] img_rgb