web-dev-qa-db-fra.com

RCNN plus rapide pour TensorFlow

Quelqu'un a-t-il implémenté FRCNN pour la version TensorFlow? J'ai trouvé des dépôts liés comme suit:

  1. Implémenter la couche de pool roi
  2. Implémenter un RCNN rapide basé sur le repo py-plus rapide-rcnn

mais pour 1: supposez que la couche de mise en commun du roi fonctionne (je n'ai pas essayé), et il y a quelque chose à implémenter comme suit:

  • Couche de données ROI, par exemple roidb .
  • Régression linéaire, par ex. SmoothL1Loss
  • Post-traitement de la couche de pool de ROI pour une formation de bout en bout qui devrait convertir les résultats de la couche de pool de ROI pour les alimenter en CNN pour le classificateur.

Pour 2: em ...., il semble basé sur py-plus rapide-rcnn qui a basé sur Caffe pour préparer le pré-traitement (par exemple roidb) et alimenter les données dans Tensorflow pour former le modèle, cela semble bizarre, donc je ne peux pas essayer il.

Donc, ce que je veux savoir, c'est que Tensorflow supportera Faster RCNN à l'avenir ?. Sinon, ai-je une mauvaise compréhension de ce qui est mentionné ci-dessus? ou un repo ou quelqu'un le soutient-il?

44
RyanLiu

Tensorflow vient de publier une API officielle de détection d'objets ici , qui peut être utilisée par exemple avec leurs différents slim modèles .

Cette API contient l'implémentation de divers pipelines pour la détection d'objets, y compris le Faster RCNN populaire, avec leurs modèles pré-formés également.

14
gdelab