puigalex / deteccion-objetos-video

Deteccion de objetos sobre video usando PyTorch
60 stars 50 forks source link

mAP compute #11

Open cvvaldez opened 3 years ago

cvvaldez commented 3 years ago

Hola Alex! me encanto tu tutorial.

Quiero calcular el mAP de mi training por pesos y estoy utilizando el siguiente comando: python train.py --model_def config/yolov3-custom.cfg --data_config config/custom.data --pretrained_weights weights/darknet53.conv.74 --batch_size 2 --compute_map True Por definición de --compute_map:

if True computes mAP every tenth batch

Estoy checando los logs cada 10 epochs y no veo los cálculos ni en los logs ni en ningun archivo que se haya generado en el repositorio. Esperaba ver algo como esto (linea #175 de train.py): print(f"---- mAP {AP.mean()}")

Sabes que podría estar pasando?