dsod测试

 

21分类测试，gpu需要45ms左右，cpu需要3.6s

 

if __name__ == '__main__':
    net = DSOD(21)
    # input = t.autograd.Variable(t.randn(1, 3, 300, 300))
    # o = m(input)
    # for ii in o:
    #     print(ii.shape)
    net=net.cuda()
    for i in range(100):
        x = torch.rand(1, 3, 416, 416)
        t1 = time.time()

        out3 = net(x.cuda())
        # print(out3)
        # print(out3)
        print(time.time() - t1)