Revert local changes

This commit is contained in:
Ivan Tikhonov 2021-12-14 06:30:45 +03:00
parent 08ffc38f1e
commit 07922f5b68
2 changed files with 97 additions and 97 deletions

View File

@ -1,98 +1,98 @@
# #
# # slice paddle model generator
# #
# import sys
# import os
#
# import numpy as np
# import paddle as pdpd
# slice paddle model generator
#
# from save_model import exportModel
# from save_model import saveModel
#
# data_type = 'float32'
#
# def slice(name : str, x, axes : list, start : list, end : list):
# pdpd.enable_static()
#
# with pdpd.static.program_guard(pdpd.static.Program(), pdpd.static.Program()):
# node_x = pdpd.static.data(name='x', shape=x.shape, dtype = data_type)
# out = pdpd.fluid.layers.slice(node_x, axes = axes, starts = start, ends = end)
#
# cpu = pdpd.static.cpu_places(1)
# exe = pdpd.static.Executor(cpu[0])
# # startup program will call initializer to initialize the parameters.
# exe.run(pdpd.static.default_startup_program())
#
# outs = exe.run(
# feed={'x': x},
# fetch_list=[out])
#
# saveModel(name, exe, feedkeys=['x'], fetchlist=[out], inputs=[x], outputs=[outs[0]], target_dir=sys.argv[1])
#
# return outs[0]
#
#
# def slice_dyn(test_shape=[2,8,10,10]):
# pdpd.disable_static()
#
# data = pdpd.rand(shape=test_shape, dtype='float32')
#
# '''
# slice w/ decrease_axis
# '''
# @pdpd.jit.to_static
# def test_slice_decrease_axis(x):
# return x[0, 1:3, :, 5]
# exportModel('slice_decrease_axis', test_slice_decrease_axis, [data], target_dir=sys.argv[1]) # output shape (2, 10)
#
# '''
# slice w/o decrease_axis
# '''
# @pdpd.jit.to_static
# def test_slice(x):
# return pdpd.slice(x, axes=[0,1,3], starts=[0,1,5], ends=[1,3,6])
# # exportModel('slice_dyn', test_slice, [data], target_dir=sys.argv[1]) # output shape (1, 2, 10, 1) # disable it by default as this kind of test model already there. It's for comparsion only.
#
# '''
# slice w/ decrease_axis of all dims
# '''
# @pdpd.jit.to_static
# def test_slice_decrease_axis_all(x):
# return x[0, 0, 0, 0]
# exportModel('slice_decrease_axis_all', test_slice_decrease_axis_all, [data], target_dir=sys.argv[1]) # output shape (1,)
#
# '''
# slice w/o decrease_axis of all dims
# '''
# @pdpd.jit.to_static
# def test_slice_alldim(x):
# return pdpd.slice(x, axes=[0,1,2,3], starts=[0,0,0,0], ends=[1,1,1,1])
# # exportModel('slice_alldim', test_slice_alldim, [data], target_dir=sys.argv[1]) # output shape (1, 1, 1, 1) # disable it by default as this kind of test model already there. It's for comparsion only.
#
# '''
# a test case simulating the last reshape2 of ocrnet which accepts slice (with decrease_axes in all dims) as its parents.
# '''
# def slice_reshape(B=1, C=256, H=16, W=32):
# pdpd.disable_static()
#
# data = pdpd.rand(shape=[B, C, H*W], dtype='float32')
#
# @pdpd.jit.to_static
# def test_model(x):
# x2 = pdpd.assign([-1, -1, 16, 32]).astype('int32')
# node_reshape = pdpd.reshape(x, [0, 256, x2[2], x2[3]])
# return node_reshape
# exportModel('slice_reshape', test_model, [data], target_dir=sys.argv[1])
#
# def main():
# x = np.linspace(1, 60, num = 60, dtype=np.int32).reshape(4, 3, 5).astype(data_type)
# slice("slice", x, axes=[1, 2], start=(0, 1), end=(-1, 3))
#
# x = np.linspace(1, 60, num = 60, dtype=np.int32).reshape(2, 30).astype(data_type)
# slice("slice_1d", x, axes=[0], start=[0], end=[1])
#
# if __name__ == "__main__":
# main()
# slice_dyn()
# slice_reshape()
import sys
import os
import numpy as np
import paddle as pdpd
from save_model import exportModel
from save_model import saveModel
data_type = 'float32'
def slice(name : str, x, axes : list, start : list, end : list):
pdpd.enable_static()
with pdpd.static.program_guard(pdpd.static.Program(), pdpd.static.Program()):
node_x = pdpd.static.data(name='x', shape=x.shape, dtype = data_type)
out = pdpd.fluid.layers.slice(node_x, axes = axes, starts = start, ends = end)
cpu = pdpd.static.cpu_places(1)
exe = pdpd.static.Executor(cpu[0])
# startup program will call initializer to initialize the parameters.
exe.run(pdpd.static.default_startup_program())
outs = exe.run(
feed={'x': x},
fetch_list=[out])
saveModel(name, exe, feedkeys=['x'], fetchlist=[out], inputs=[x], outputs=[outs[0]], target_dir=sys.argv[1])
return outs[0]
def slice_dyn(test_shape=[2,8,10,10]):
pdpd.disable_static()
data = pdpd.rand(shape=test_shape, dtype='float32')
'''
slice w/ decrease_axis
'''
@pdpd.jit.to_static
def test_slice_decrease_axis(x):
return x[0, 1:3, :, 5]
exportModel('slice_decrease_axis', test_slice_decrease_axis, [data], target_dir=sys.argv[1]) # output shape (2, 10)
'''
slice w/o decrease_axis
'''
@pdpd.jit.to_static
def test_slice(x):
return pdpd.slice(x, axes=[0,1,3], starts=[0,1,5], ends=[1,3,6])
# exportModel('slice_dyn', test_slice, [data], target_dir=sys.argv[1]) # output shape (1, 2, 10, 1) # disable it by default as this kind of test model already there. It's for comparsion only.
'''
slice w/ decrease_axis of all dims
'''
@pdpd.jit.to_static
def test_slice_decrease_axis_all(x):
return x[0, 0, 0, 0]
exportModel('slice_decrease_axis_all', test_slice_decrease_axis_all, [data], target_dir=sys.argv[1]) # output shape (1,)
'''
slice w/o decrease_axis of all dims
'''
@pdpd.jit.to_static
def test_slice_alldim(x):
return pdpd.slice(x, axes=[0,1,2,3], starts=[0,0,0,0], ends=[1,1,1,1])
# exportModel('slice_alldim', test_slice_alldim, [data], target_dir=sys.argv[1]) # output shape (1, 1, 1, 1) # disable it by default as this kind of test model already there. It's for comparsion only.
'''
a test case simulating the last reshape2 of ocrnet which accepts slice (with decrease_axes in all dims) as its parents.
'''
def slice_reshape(B=1, C=256, H=16, W=32):
pdpd.disable_static()
data = pdpd.rand(shape=[B, C, H*W], dtype='float32')
@pdpd.jit.to_static
def test_model(x):
x2 = pdpd.assign([-1, -1, 16, 32]).astype('int32')
node_reshape = pdpd.reshape(x, [0, 256, x2[2], x2[3]])
return node_reshape
exportModel('slice_reshape', test_model, [data], target_dir=sys.argv[1])
def main():
x = np.linspace(1, 60, num = 60, dtype=np.int32).reshape(4, 3, 5).astype(data_type)
slice("slice", x, axes=[1, 2], start=(0, 1), end=(-1, 3))
x = np.linspace(1, 60, num = 60, dtype=np.int32).reshape(2, 30).astype(data_type)
slice("slice_1d", x, axes=[0], start=[0], end=[1])
if __name__ == "__main__":
main()
slice_dyn()
slice_reshape()

View File

@ -2,5 +2,5 @@
# SPDX-License-Identifier: Apache-2.0
telemetry_params = {
'TID': "0"
'TID': "UA-17808594-29"
}