2016-07-18 114 views
8

È possibile avere previsioni di bounding box utilizzando TensorFlow? Ho trovato TensorBox su github ma sto cercando un modo migliore supportato o forse ufficiale per risolvere questo problema.Scatole di contenimento con tensorflow e inizio-v3

Ho bisogno di riqualificare il modello per le mie classi.

+0

Hi @BernardoGO Ho lo stesso problema, si sono trovate qualche soluzione per questo? –

risposta

5

Non è chiaro cosa intendi esattamente. Hai bisogno di rilevamento oggetti? Lo presumo dalle 'scatole di delimitazione'. In tal caso, le reti iniziali non sono direttamente applicabili al tuo compito, sono reti di classificazione. Dovresti cercare modelli di rilevamento oggetti, come Single Shot Detector (SSD) o You Only Look Once (YOLO). Spesso usano strati convoluzionali pre-addestrati da reti di classificazione, ma hanno strati aggiuntivi in ​​cima. Se vuoi Inception (ovvero GoogLeNet), YOLO si basa su questo. Dai un'occhiata a questa implementazione: https://github.com/thtrieu/darkflow o qualsiasi altra che puoi trovare su Google.

+0

Hai provato Darkflow? Considero la documentazione terribile. –

+0

No, stavo usando darknet, anche documentato principalmente con RTFS ... –

1

Il vincitore di COCO2016 per il rilevamento di oggetti è stato implementato in tensorflow. Alcune tecniche avanzate sono Faster R-CNN, R-FCN e SSD. Controlla le diapositive da http://image-net.org/challenges/talks/2016/GRMI-COCO-slidedeck.pdf (La diapositiva 14 ha delle chiavi tensorflow op che ti consentono di ricreare questa pipeline).

Edit 2017/06/19: tensorflow rilasciato alcune tecniche per prevedere bboxes: https://research.googleblog.com/2017/06/supercharge-your-computer-vision-models.html