Commit ccf9a521 authored by Yuxin Wu's avatar Yuxin Wu

fix pep8 style in /scripts

parent 9b69e860
...@@ -7,8 +7,6 @@ ...@@ -7,8 +7,6 @@
import numpy as np import numpy as np
from tensorpack.tfutils.varmanip import dump_chkpt_vars from tensorpack.tfutils.varmanip import dump_chkpt_vars
from tensorpack.utils import logger from tensorpack.utils import logger
import tensorflow as tf
import sys
import argparse import argparse
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
...@@ -26,4 +24,5 @@ logger.info(str(params.keys())) ...@@ -26,4 +24,5 @@ logger.info(str(params.keys()))
if args.dump: if args.dump:
np.save(args.dump, params) np.save(args.dump, params)
if args.shell: if args.shell:
import IPython as IP; IP.embed(config=IP.terminal.ipapp.load_default_config()) import IPython as IP
IP.embed(config=IP.terminal.ipapp.load_default_config())
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: dump_train_config.py # File: dump-dataflow.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com> # Author: Yuxin Wu <ppwwyyxx@gmail.com>
import argparse import argparse
import cv2 import cv2
import tensorflow as tf
import imp import imp
import tqdm import tqdm
import os import os
from tensorpack.utils import logger from tensorpack.utils import logger
from tensorpack.utils.fs import mkdir_p from tensorpack.utils.fs import mkdir_p
from tensorpack.dataflow import * from tensorpack.dataflow import RepeatedData
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
...@@ -54,6 +53,3 @@ with tqdm.tqdm(total=NR_DP_TEST, leave=True, unit='data points') as pbar: ...@@ -54,6 +53,3 @@ with tqdm.tqdm(total=NR_DP_TEST, leave=True, unit='data points') as pbar:
if idx > NR_DP_TEST: if idx > NR_DP_TEST:
break break
pbar.update() pbar.update()
...@@ -8,8 +8,9 @@ import argparse ...@@ -8,8 +8,9 @@ import argparse
import tensorflow as tf import tensorflow as tf
import imp import imp
from tensorpack import * from tensorpack import TowerContext, logger, ModelFromMetaGraph
from tensorpack.tfutils import sessinit, varmanip from tensorpack.tfutils import sessinit, varmanip
from tensorpack.utils.naming import EXTRA_SAVE_VARS_KEY
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--config', help='config file') parser.add_argument('--config', help='config file')
...@@ -51,6 +52,6 @@ with tf.Graph().as_default() as G: ...@@ -51,6 +52,6 @@ with tf.Graph().as_default() as G:
logger.info("Variables to dump:") logger.info("Variables to dump:")
logger.info(", ".join(var_dict.keys())) logger.info(", ".join(var_dict.keys()))
saver = tf.train.Saver( saver = tf.train.Saver(
var_list=var_dict, var_list=var_dict,
write_version=tf.train.SaverDef.V2) write_version=tf.train.SaverDef.V2)
saver.save(sess, args.output, write_meta_graph=False) saver.save(sess, args.output, write_meta_graph=False)
This diff is collapsed.
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: serve_data.py # File: serve-data.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com> # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import argparse import argparse
import imp import imp
#import cv2
#import os
from tensorpack.dataflow import serve_data from tensorpack.dataflow import serve_data
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment