Faster-RCNN_TF的loss曲线可视化（tensorflow版本，Faster RCNN的loss曲线可视化）

最新推荐文章于 2026-06-27 07:05:03 发布

原创

最新推荐文章于 2026-06-27 07:05:03 发布 · 9.5k 阅读

收录于

当前文章被以下社区和专栏收录：

本文介绍了如何在TensorFlow实现的Faster-RCNN中，通过修改train.py文件，每cfg.TRAIN.SNAPSHOT_ITERS步保存loss数据到csv，然后读取这些数据绘制loss曲线。作者分享了具体的代码修改和画图步骤，并展示了最终的all_loss曲线图，讨论了可能的learning rate设置问题。

我用的Faster-RCNN是tensorflow版本，github地址：Faster-RCNN_TF
查到网上的loss曲线都是直接从log的txt里读取，我的代码不生成txt，所以得自己改。我的思路是每cfg.TRAIN.SNAPSHOT_ITERS步生成一个csv文件，最后读取这个文件再画图。

具体操作如下：

1. 修改train.py

在/lib/fast rcnn/train.py中，train_model(self, sess, max_iters)函数里：
在for iter in range(max_iters):这句的上下对应修改，改后代码如下：

        ####add
        all_loss = []
        rpn_cls_loss = []
        rpn_box_loss = []
        cls_loss = []
        box_loss = []
        for iter in range(max_iters):
            # get one batch
            blobs = data_layer.forward()

            # Make one SGD update
            feed_dict={
  
  self.net.data: blobs['data'], self.net.im_info: blobs['im_info'], self.net.keep_prob: 0.5, \
                           self.net.gt_boxes: blobs['gt_boxes']}

            ru