Facilitando a troca de arquitetura e ajustando README

README.md

@@ -22,6 +22,16 @@ Driver nvidia: 450.66
 cd ./utils/
 gedit class_names.py faster_rcnn_r50_fpn_1x_voc0712.py voc.py
 
+### Usando outra arquitetura
+# 
+# - Troque o nome do arquivo de configuração da arquitetura (sem o .py) no arquivo arquitetura.txt
+# - Opções disponíveis até agora (ajuste outros arquivos disponíveis em mmdetection/configs para
+#   outras arquiteturas)
+#   * faster_rcnn_r50_fpn_1x_voc0712
+#   * retinanet_r50_fpn_1x_voc0712
+cd ./run
+gedit arquitetura.txt
+
 ### Exemplo de uso:
 # 
 # - Use o labelImg para anotar as imagens no formato pascalVOC

run/arquitetura.txt

+faster_rcnn_r50_fpn_1x_voc0712

run/detecta.sh

@@ -4,7 +4,7 @@
 # - É preciso treinar a rede antes e trocar
 #   o modelo usado se necessário
 
-modelo=faster_rcnn_r50_fpn_1x_voc0712
+modelo=`cat arquitetura.txt`
 config=configs/pascal_voc/${modelo}.py
 pesos=work_dirs/${modelo}/latest.pth
 

run/testa.sh

@@ -4,7 +4,7 @@
 # - Se necessário troque o nome do modelo
 #   na linha de baixo se não usou a faster_RCNN
 
-modelo=faster_rcnn_r50_fpn_1x_voc0712
+modelo=`cat arquitetura.txt`
 config=configs/pascal_voc/${modelo}.py
 pesos=work_dirs/${modelo}/latest.pth
 

run/treina.sh

@@ -5,7 +5,7 @@
 
 
 
-modelo=faster_rcnn_r50_fpn_1x_voc0712
+modelo=`cat arquitetura.txt`
 config=configs/pascal_voc/${modelo}.py
 pesos=work_dirs/${modelo}/latest.pth
 

utils/retinanet_r50_fpn_1x_voc0712.py

@@ -17,4 +17,4 @@ optimizer_config = dict(grad_clip=None)
 # actual epoch = 3 * 3 = 9
 lr_config = dict(policy='step', step=[3])
 # runtime settings
-total_epochs = 40  # actual epoch = 4 * 3 = 12
+total_epochs = 1000