找回密码
 立即注册
  • TA的每日心情
    难过
    2016-7-4 14:24
  • 签到天数: 3 天

    [LV.2]偶尔看看I

    发表于 2018-3-6 13:54:17
    1131
    Creating DetectNet Model with DIGITS
    当上一步的数据处理完,返回到主页上面。选择Models,然后选择NewModel里面的Object Detection:
    按照下面的内容设置你的表单中的内容:

    • Select Dataset: coco-dog
    • Training epochs: 100
    • Subtract Mean: none
    • Solver Type: Adam
    • Base learning rate: 2.5e-05
    • Select Show advanced learning options

          
      • Policy: Exponential Decay
          
      • Gamma: 0.99
    Selecting DetectNet Batch Size
    DetectNet的默认设置batchsize是10,训练时会消耗大概12GB的内存空间。利用 Batch Accumulation 方法,可以在一个GPU上进行训练,不用占用12GB那么多的内存空间。下面的表格可以查看对应的BatchSize占用的GPU内存空间。
            
    GPU Memory
             
    Batch Size
          
    Batch Accumulation
         
       
    4GB
       
    2
       
    5
       
       
    8GB
       
    5
       
    2
       
       
    12GB or  larger
       
    [network defaults] (10)
       
    Leave blank (1)
       















    如果您的GPU卡有12GB,您可直接设置默认大小的Batch Size而不用设置 Batch Accumulation 。如果您想用更少的GPU或者GPU内存,您可以按照上面的进行设置。
    Specifying the DetectNet Prototxt
    在网络模型定义方面,我们选择Custom Network 来自定义修改网络。在我们的例子中,我们可以复制粘贴detectnet.prototxt里面的内容。
    DetecNet prototxt 在data/networks/detectnet.prototxt
    Training the Model with Pretrained Googlenet
    因为DetectNet是有GoogleNet衍生出来的,所以强烈推荐使用GoogleNet作为预训练权重,这样会保证训练的速度和稳定。下载GoogleNet模型,或者用下面的命令下载:
    然后在Prertrained Model 选项里指定下载的GoogleNet的路径。
    选择一个GPU来训练,然后填好Model Name和Group Name:

    • Group Name MS-COCO
    • Model Name DetectNet-COCO-Dog
    最后,点击页面最下面的Create按钮:
    Testing DetectNet Model Inference in DIGITS
    让训练工作跑一段时间,比如50个epochs,直到mAP(Mean Average Precision)的曲线开始上升。注意,由于mAP的计算室友DetecNet的损失函数完成的,所以mAP的变化区域不一定是0~100,就算是5或者10,这个模型也是可以用的。在训练COCO这个量级的数据的时候,可能需要利用GPU跑几个小时才能完成。
    训练完成之后,我们可以再训练的这个页面测试一下我们训练好的模型。在这个页面,我们看最下面,在Visualization里面选择Bounding boxes,然后在下面的Image Path里面输入您想要测试的图片(我们这里的例子是:/coco/val/images/dog/000074.png):
    然后点击测试按钮,您就会看到如下界面:

    使用道具 举报 回复
    发表于 2018-4-16 20:48:16
    楼主有没有发现,验证的时候,不同的batchsize,detectnet计算出来的mAP、recall、precision不同?
    使用道具 举报 回复 支持 反对
    发新帖
    您需要登录后才可以回帖 登录 | 立即注册

    zzczczxczxczx