Search code examples
tensorflowdeep-learningtensorboardobject-detection-apiloss

Plot validation loss in Tensorflow Object Detection API


I'm using Tensorflow Object Detection API for detection and localization of one class object in images. For these purposes, I use the pre-trained faster_rcnn_resnet50_coco_2018_01_28 model.

I want to detect under/overfitting after training the model. I see training loss, but after evaluating Tensorboard only shows mAP and Precision metrics and no loss.

Is this possible to plot a validation loss on Tensorboard too?


Solution

  • There is validation loss. Assuming you're using the latest API, the curve under "loss" is validation loss while "loss_1/2" is the training loss.