kentaroy47 / frcnn-from-scratch-with-keras

:collision:Faster R-CNN from scratch written with Keras
Apache License 2.0
168 stars 106 forks source link

Metrics monitoring #16

Open palarax opened 5 years ago

palarax commented 5 years ago

Is it possible to add Tensorboard to allow for monitoring

This is what I've done in "train_rpn.py" but i only seem to see Graph

model_rpn.compile(optimizer=optimizer, loss=[losses.rpn_loss_cls(num_anchors), losses.rpn_loss_regr(num_anchors)],metrics=['accuracy']) ... Tensorboard = TensorBoard(log_dir='logs/{}'.format(time())) callback=[Callbacks, Tensorboard] history = model_rpn.fit_generator(data_gen_train,epochs=options.num_epochs, validation_data=data_gen_val,steps_per_epoch=1000,callbacks=callback, validation_steps=10)

I can see rpn loss history is saved, but how can we graph it

import numpy numpy_loss_history = numpy.array(loss_history) numpy.savetxt(options.network+"_rpn_loss_history.txt", numpy_loss_history, delimiter=",")

kentaroy47 commented 5 years ago

ok, let me go through this.

mariaculman18 commented 5 years ago

Hi @kentaroy47, thank you for making available this project.

This ongoing on metrics monitoring means the mAP estimation would be included in the tensorboard? Is there a present way to measure mAP? Is there a present way to measure loss with a validation dataset?