示例#1
0
def test_to_from_bytes(image):
    net = Network.load('tiny-yolo')
    data = net.to_bytes()
    results = net(image)
    loaded = Network().from_bytes(data)
    results2 = net(image)
    assert results == results2
示例#2
0
def test_init():
    nn = Network()
示例#3
0
def test_from_disk(image):
    net = Network().from_disk(Path('lightnet/data/tiny-yolo').resolve())
    results = net(image)
    assert results is not None