use_voxelchain.py 文件源码

def use_model(model):

    data = np.loadtxt('data/human_test_1_32_32_32.txt').reshape(1, 1, 32, 32, 32).astype(np.float32)
    y= model.fwd(data)
    A= F.softmax(y).data
    print(A.argmax(axis=1))
    print(A[0,A.argmax(axis=1)])