def test_InceptionV1Structure(): n_classes = 10 # model_types = [1, 2, 3, 4] model_types = [1] for model_type in model_types: print(f'test_model_type= {model_type}') x = tf.placeholder(tf.float32, [None, 128, 128, 3]) net = InceptionV1NetModule(x, n_classes, model_type=model_type) with elapse_time('build'): net.build() x_np = np.random.normal(0, 1, [10, 128, 128, 3]) with tf.Session() as sess: sess.run(tf.global_variables_initializer()) with elapse_time('net proba forward time'): forward = sess.run(net.proba, feed_dict={x: x_np}) print(forward.shape) with elapse_time('aux0 proba forward time'): forward = sess.run(net.aux0_proba, feed_dict={x: x_np}) print(forward.shape) with elapse_time('aux1 proba forward time'): forward = sess.run(net.aux1_proba, feed_dict={x: x_np}) print(forward.shape)
def test_ResNet101Structure(): x = tf.placeholder(tf.float32, [None, 128, 128, 3]) n_classes = 10 net = ResNet101Structure(x, n_classes) with elapse_time('build'): net.build() h = net.proba x_np = np.random.normal(0, 1, [10, 128, 128, 3]) with tf.Session() as sess: sess.run(tf.global_variables_initializer()) with elapse_time('forward time'): forward = sess.run(h, feed_dict={x: x_np}) print(forward.shape)
def test_NpSharedObj(): shape = [3, 3] mat = np.ones(shape) shared_obj = NpSharedObj.from_np(mat) print(shared_obj.np) print(f'share memory') with Pool(processes=1, initializer=init_shared, initargs=(shared_obj.loc, shape,)) as pool: with elapse_time(): childs = [pool.apply_async(func2, args=(args,)) for args in range(16)] for child in childs: a = child.get() print(a) print(f'exit') print(shared_obj.np)