Comment puis-je obtenir des boîtes détectées sur l'ensemble d'évaluation pour l'API de détection d'objet de tensorflow?

Question

Je travaillais avec l'API Tensorflow récemment publiée pour la détection d'objets, avec le RCNN plus rapide sur Resnet 101 sur mon propre ensemble de données. Il semble s'entraîner et évaluer sur les données de validation, mais j'espérais qu'il y avait un moyen d'obtenir/stocker des bounding boxes pour toutes les images du set Eval, dans un fichier, ou peut-être, obtenir l'emplacement dans le code source peut obtenir les boîtes englobantes prédites avec des noms d'image.

Answer 1

Si vous souhaitez simplement obtenir les cadres de délimitation détectés en fonction d'un ensemble d'images, le bloc-notes de Jupyter contient un bon exemple de la façon de procéder.