[Files] load and assign model from npz

Author: zsdonghao

docs/modules/files.rst

@@ -44,6 +44,7 @@ sake of cross-platform.
    save_npz
    load_npz
    assign_params
+   load_and_assign_npz
 
    save_any_to_npy
    load_npy_to_any
@@ -103,6 +104,10 @@ Assign parameters to network
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 .. autofunction:: assign_params
 
+Load and assign parameters to network
+^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+.. autofunction:: load_and_assign_npz
+
 Load and save variables
 ------------------------
 

tensorlayer/files.py

@@ -634,7 +634,31 @@ def assign_params(sess, params, network):
         ops.append(network.all_params[idx].assign(param))
     sess.run(ops)
 
+def load_and_assign_npz(sess=None, name=None, network=None):
+    """Load model from npz and assign to a network.
 
+    Parameters
+    -------------
+    sess : TensorFlow Session
+    name : string
+        Model path.
+    network : a :class:`Layer` class
+        The network to be assigned
+
+    Examples
+    ---------
+    >>> tl.files.load_and_assign_npz(sess=sess, name='net.npz', network=net)
+    """
+    assert network is not None
+    assert sess is not None
+    if not os.path.exists(name):
+        print("[!] Load {} failed!".format(name))
+        return False
+    else:
+        params = tl.files.load_npz(name=name)
+        tl.files.assign_params(sess, params, network)
+        print("[*] Load {} SUCCESS!".format(name))
+        return network
 
 # Load and save variables
 def save_any_to_npy(save_dict={}, name='any.npy'):