Commit ccf9a521 authored by Yuxin Wu's avatar Yuxin Wu

fix pep8 style in /scripts

parent 9b69e860
......@@ -7,8 +7,6 @@
import numpy as np
from tensorpack.tfutils.varmanip import dump_chkpt_vars
from tensorpack.utils import logger
import tensorflow as tf
import sys
import argparse
parser = argparse.ArgumentParser()
......@@ -26,4 +24,5 @@ logger.info(str(params.keys()))
if args.dump:
np.save(args.dump, params)
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
# -*- coding: UTF-8 -*-
# File: dump_train_config.py
# File: dump-dataflow.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import argparse
import cv2
import tensorflow as tf
import imp
import tqdm
import os
from tensorpack.utils import logger
from tensorpack.utils.fs import mkdir_p
from tensorpack.dataflow import *
from tensorpack.dataflow import RepeatedData
parser = argparse.ArgumentParser()
......@@ -54,6 +53,3 @@ with tqdm.tqdm(total=NR_DP_TEST, leave=True, unit='data points') as pbar:
if idx > NR_DP_TEST:
break
pbar.update()
......@@ -8,8 +8,9 @@ import argparse
import tensorflow as tf
import imp
from tensorpack import *
from tensorpack import TowerContext, logger, ModelFromMetaGraph
from tensorpack.tfutils import sessinit, varmanip
from tensorpack.utils.naming import EXTRA_SAVE_VARS_KEY
parser = argparse.ArgumentParser()
parser.add_argument('--config', help='config file')
......@@ -51,6 +52,6 @@ with tf.Graph().as_default() as G:
logger.info("Variables to dump:")
logger.info(", ".join(var_dict.keys()))
saver = tf.train.Saver(
var_list=var_dict,
write_version=tf.train.SaverDef.V2)
var_list=var_dict,
write_version=tf.train.SaverDef.V2)
saver.save(sess, args.output, write_meta_graph=False)
This diff is collapsed.
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: serve_data.py
# File: serve-data.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import argparse
import imp
#import cv2
#import os
from tensorpack.dataflow import serve_data
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