Herramientas desarrolladas en el marco del desaf铆o "IA por la Identidad" organizado por la Fundaci贸n Sadosky, el Ministerio de Ciencia, Tecnolog铆a e Innovaci贸n y Abuelas de Plaza de Mayo (馃 2do Lugar)
Solicitar acceso al repositorio principal | Descargar el modelo y dataset | Integrantes del equipo | Notas de prensa
Requiere una instalaci贸n de Miniconda o alguna variante de Miniforge.
$ conda env create -f environment.yml
$ conda activate diaxi-training-tools
-
notebooks/create_detectron2_model.ipynb
En esta notebook se explica c贸mo entrenamos el modelo de segmentaci贸n que luego implementamos en el software. Fue pensada para ejecutarse en Google Colab ya que no cont谩bamos con una GPU.
-
notebooks/create_diaxi_dataset.ipynb
En esta notebook se detalla paso a paso c贸mo se cre贸 el dataset en formato COCO y los problemas que fuimos solucionando en el camino.
-
scripts/bbox2mask.py
Este script crea m谩scaras de segmentaci贸n rectangulares para datasets en formato COCO que solamente contienen bounding boxes.
-
scripts/catfilter.py
Este script permite descartar las etiquetas que no queremos utilizar para entrenar el modelo.
-
scripts/imgidfix.py
Este script mapea las IDs de las im谩genes entre
0..N
. Es necesario para fusionar datasets conpyodi
.