rfcn训练记录。

1、如果需要修改模型的存储位置。更改py-R-FCN/lib/fast_rcnn/config.py下的:
  outdir = osp.abspath(osp.join(__C.ROOT_DIR, 'output_big_finetuning', __C.EXP_DIR, imdb.name))
2、如果训练的类别不同。更改prototxt。
3、如果训练模型的感受野/feature map大小发生变化。
A.更改prototxt里的'feat_stride': 16"和spatial_scale: 0.0625。
B./home/cowa/lhb/py-R-FCN/lib/rpn/generate.py
def imdb_rpn_compute_stats(net, imdb, anchor_scales=(8,16,32),
                           feature_stride=16):
C./home/cowa/lhb/py-R-FCN/lib/rpn/generate_anchors.py
def generate_anchors(base_size=16, ratios=[0.5, 1, 2],
                     scales=2**np.arange(3, 6)):
    """
4、如果需要蒋原本的数据弄成两类进行训练,修改:
A./home/cowa/lhb/py-R-FCN/lib/datasets/pascal_voc.py
self._classes = ('__background__', # always index 0
                         'red', 'yellow', 'green', 'off')
同时:
            #cls='traffic_light'
            #cls = self._class_to_ind['traffic_light']
            cls = self._class_to_ind[obj.find('name').text.lower().strip()]
B./home/cowa/lhb/py-R-FCN/lib/datasets/voc_eval.py
#R = [obj for obj in recs[imagename] if obj['name'] == 'red' or obj['name'] == 'off' or obj['name'] == 'yellow' or obj['name'] == 'green']
        R = [obj for obj in recs[imagename] if obj['name'] ==classname]

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值