RRPN/R2CNN实验代码修改记录

本文详细记录了RRPN/R2CNN实验过程中对代码的修改,包括显卡配置、tfrecord文件生成、错误处理以及针对ICDAR2015数据集的适应性调整。在处理opencv版本问题和XML格式错误后,成功生成tfrecord文件,并通过调整cfgs.py文件参数进行训练。同时,介绍了如何使用tensorboard监测训练过程。
摘要由CSDN通过智能技术生成
  1. 修改显卡:cfgs.py文件 第15行 GPU_GROUP=“1”
    查看显卡使用情况: nvidia-smi

  2. 生成tfrecord文件:
    -a: 改cfgs.py
    第9行 NET_NAME=‘resnet_v1_50’ 作者用的是101
    第8行 VERSION=‘RRPN_VOCdevkit2007(MLT)_v1’
    第64行 DATASET_NAME=‘VOCdevkit2007(MLT)’
    CLASS_NUM改为1 试一试
    -b: 在/libs/label_dict.py里面添加dataset

elif cfgs.DATASET_NAME=='VOCdevkit2007(MLT)':
      NAME_LABEL_MAP={
   
             'background' : 0
  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值