解释下面代码num_workers = 0 pretrained_weights = 'checkpoints/v3_ctpn_ep22_0.3801_0.0971_0.4773.pth' anchor_scale = 16 IOU_NEGATIVE = 0.3 IOU_POSITIVE = 0.7 IOU_SELECT = 0.7 RPN_POSITIVE_NUM = 150 RPN_TOTAL_NUM = 300 # bgr can find from here: https://github.com/fchollet/deep-learning-models/blob/master/imagenet_utils.py IMAGE_MEAN = [123.68, 116.779, 103.939] OHEM = True checkpoints_dir = './checkpoints' outputs = r'./logs'
时间: 2023-05-12 21:06:25 浏览: 293
deep-learning-with-python-notebooks-master
num_workers = 0:表示使用当前计算机的所有CPU核心进行训练。
pretrained_weights = 'checkpoints/v3_ctpn_ep22_0.3801_0.0971_0.4773.pth':表示使用预训练的权重文件进行模型的初始化。
anchor_scale = 16:表示使用的anchor的大小比例。
IOU_NEGATIVE = 0.3:表示IOU小于0.3的anchor被认为是负样本。
IOU_POSITIVE = 0.7:表示IOU大于0.7的anchor被认为是正样本。
IOU_SELECT = 0.7:表示在训练时选择的anchor的IOU阈值。
RPN_POSITIVE_NUM = 150:表示每张图片中正样本的数量。
RPN_TOTAL_NUM = 300:表示每张图片中总共选择的anchor的数量。
阅读全文