예제 #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