更快的 r-cnn 测试文件夹中的所有图像

Question

我在计算机上使用 fast-rcnn 训练了我的模型。如您所知，使用 脚本只能测试 1 个图像。但我想测试测试文件夹中的所有图像。我必须编写一个代码（使用循环）来使我能够做到这一点。有谁可以帮助我解决这个问题。提前致谢。真挚地。

Answer 1

你好，
在这里你可以找到我实现的用于测试的函数
一组图像上的 Faster R-CNN：

def get_folder_results(detector, image_dir, device):
       ' The detector represents your Faster R-CNN model,
         image_dir represents the folderpath containing the images,
         device : the device used to train (CPU, GPU ...) '
 for image in os.listdir(image_dir):
    image_path = os.path.join(image_dir, image)
    input_images = [T.ToTensor()(Image.open(image_path)).to(device)]
    prediction = detector(images)
    print(prediction)

    

执行此函数后，您将获得每个图像的预测，其中每个
预测表示边界框的坐标和
相应的类）。