Search code examples
pythontensorflowobject-detectionobject-detection-api

Tensorflow Object Detection API only shows loss value. How to get mAP value?


I am training my own dataset using Tensorflow Object Detection API. While training, I only get loss value like this. Is there any way to get mAP value while training or on tensorboard ?


Solution

  • in the legacy/train.py and legacy/eval.py

    Example usage:
        ./train \
            --logtostderr \
            --train_dir=path/to/train_dir \
            --pipeline_config_path=pipeline_config.pbtxt
    
    Example usage:
        ./eval \
            --logtostderr \
            --checkpoint_dir=path/to/checkpoint_dir \
            --eval_dir=path/to/eval_dir \
            --pipeline_config_path=pipeline_config_path.pbtxt \
    

    for one terminal you use train.py to train ,

    for another terminal you use eval.py to evaluate ,

    make sure train_dir same as checkpoint_dir and eval_dir

    and the other terminal open up tensorboard

    make sure u arrange ur hardware device, otherwise u might not able open both train and eval in the same time