Commit 292b27cf authored by Hemerson Pistori's avatar Hemerson Pistori
Browse files

Facilitando a troca de arquitetura e ajustando README

parent 0cac1e76
......@@ -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
......
faster_rcnn_r50_fpn_1x_voc0712
......@@ -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
......
......@@ -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
......
......@@ -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
......
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment