Detectron: Comment entraîner l'ensemble de données personnalisé.
Salut! Tout d'abord, merci pour vos merveilleux travaux.
J'ai fini d'entraîner le jeu de données RetinaNet avec COCO comme vous l'avez demandé.
Je souhaite entraîner mon propre ensemble de données avec RetinaNet ou un autre modèle de base.
Je regarde la structure du code interne et je constate que toutes les configurations de modèle sont définies dans le fichier * .yaml et train_net.py lit le fichier * .yaml et construit la base de données à partir du fichier .jason dans le répertoire d'annotations COCO.
Donc, si je veux entraîner mon propre ensemble de données, le seul moyen est de générer le fichier .json similaire avec les annotations COCO?
nonstop1962
Tous les 5 commentaires
@ nonstop1962 : oui, la méthode recommandée est de convertir votre jeu de données au format d'annotation COCO json. Pour les boîtes englobantes, cela peut généralement être fait en <100 lignes de Pythons. Bien sûr, vous pouvez apporter des modifications arbitraires au code Detectron pour prendre en charge les formats personnalisés, mais c'est probablement plus difficile et plus sujet à des problèmes avec des cas d'angle manqués.
rbgirshick
le 25 janv. 2018
Merci pour votre réponse!
nonstop1962
le 25 janv. 2018
Merci pour @rbgirshick !
Ok, alors puis-je obtenir un ensemble de données d'entraînement si je souhaite segmenter un t-shirt?
par exemple, JSON get pour les utilisations de l'API COCO.
Je pense que cette plateforme peut segmenter.
d'accord??
Merci ~!
oryondark
le 19 sept. 2018
Salut,
pour ceux qui en ont besoin, voici un script pour convertir les annotations xml pascal voc au format coco json: https://github.com/gamcoh/Object-Detection-Tools/blob/master/pascal_voc_xml2coco_json.py
gamcoh
le 20 mars 2019
Merci gamcoh
oryondark
le 21 mars 2019
Questions connexes
Emma0928
·
3Commentaires
pacelu
·
3Commentaires
twmht
·
3Commentaires
junxiaoge
·
3Commentaires
coldgemini
·
3Commentaires
Commentaire le plus utile
@ nonstop1962 : oui, la méthode recommandée est de convertir votre jeu de données au format d'annotation COCO json. Pour les boîtes englobantes, cela peut généralement être fait en <100 lignes de Pythons. Bien sûr, vous pouvez apporter des modifications arbitraires au code Detectron pour prendre en charge les formats personnalisés, mais c'est probablement plus difficile et plus sujet à des problèmes avec des cas d'angle manqués.