Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

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?

like image 956
Ruslan Avatar asked Jun 10 '26 09:06

Ruslan


2 Answers

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.

like image 200
netanel-sam Avatar answered Jun 11 '26 23:06

netanel-sam


To see the validation curve you should change faster_rcnn_resnet50_coco.config:

1- comment max_evals line
2- set eval_interval_secs: 60 .
3- num_examples should be equal or less than the number of "files" that you have in "val.record" .

eval_config: { . 
  num_examples: 600 . 
  eval_interval_secs: 60 . 
  # Note: The below line limits the evaluation process to 10 evaluations.  
  # Remove the below line to evaluate indefinitely.  
  # max_evals: 10 .
}
like image 41
Marjan Moodi Avatar answered Jun 11 '26 22:06

Marjan Moodi



Donate For Us

If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!