Python resnetの例

コード例 #1

ファイル: test_resnet.py プロジェクト: pankajkarman/leibniz

 def testHyp3D(self):
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  block=HyperBasic)
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  normalizor='instance',
                  block=HyperBasic)
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  normalizor='layer',
                  block=HyperBasic)
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  block=HyperBottleneck)
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  normalizor='instance',
                  block=HyperBottleneck)
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  normalizor='layer',
                  block=HyperBottleneck)
     net(th.rand(1, 1, 4, 16, 32))
     net(th.rand(2, 1, 4, 16, 32))

コード例 #2

ファイル: test_resnet.py プロジェクト: caiyunapp/leibniz

 def test3D(self):
     resnet(1, 1, spatial=(16, 16, 16))
     resnet(1, 1, spatial=(32, 16, 16))
     resnet(1, 1, spatial=(16, 32, 16))
     resnet(1, 1, spatial=(16, 16, 32))
     resnet(1, 1, spatial=(11, 16, 32))
     net = resnet(1, 1, spatial=(4, 16, 32))
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1, 1, spatial=(4, 16, 32), normalizor='instance')
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1, 1, spatial=(4, 16, 32), normalizor='layer')
     net(th.rand(1, 1, 4, 16, 32))
     net(th.rand(2, 1, 4, 16, 32))

コード例 #3

ファイル: test_resnet.py プロジェクト: pankajkarman/leibniz

 def test3D(self):
     resnet(1, 1, spatial=(16, 16, 16))
     resnet(1, 1, spatial=(32, 16, 16))
     resnet(1, 1, spatial=(16, 32, 16))
     resnet(1, 1, spatial=(16, 16, 32))
     resnet(1, 1, spatial=(11, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]])
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  normalizor='instance')
     net(th.rand(1, 1, 4, 16, 32))
     net = resnet(1,
                  1,
                  spatial=(4, 16, 32),
                  scales=[[0, -1, -1], [-1, -1, -1], [0, -1, -1],
                          [-1, -1, -1]],
                  normalizor='layer')
     net(th.rand(1, 1, 4, 16, 32))
     net(th.rand(2, 1, 4, 16, 32))