(self, FLAGS, darknet = None)
| 34 | load_from_ckpt = help.load_from_ckpt |
| 35 | |
| 36 | def __init__(self, FLAGS, darknet = None): |
| 37 | self.ntrain = 0 |
| 38 | |
| 39 | if isinstance(FLAGS, dict): |
| 40 | from ..defaults import argHandler |
| 41 | newFLAGS = argHandler() |
| 42 | newFLAGS.setDefaults() |
| 43 | newFLAGS.update(FLAGS) |
| 44 | FLAGS = newFLAGS |
| 45 | |
| 46 | self.FLAGS = FLAGS |
| 47 | if self.FLAGS.pbLoad and self.FLAGS.metaLoad: |
| 48 | self.say('\nLoading from .pb and .meta') |
| 49 | self.graph = tf.Graph() |
| 50 | device_name = FLAGS.gpuName \ |
| 51 | if FLAGS.gpu > 0.0 else None |
| 52 | with tf.device(device_name): |
| 53 | with self.graph.as_default() as g: |
| 54 | self.build_from_pb() |
| 55 | return |
| 56 | |
| 57 | if darknet is None: |
| 58 | darknet = Darknet(FLAGS) |
| 59 | self.ntrain = len(darknet.layers) |
| 60 | |
| 61 | self.darknet = darknet |
| 62 | args = [darknet.meta, FLAGS] |
| 63 | self.num_layer = len(darknet.layers) |
| 64 | self.framework = create_framework(*args) |
| 65 | |
| 66 | self.meta = darknet.meta |
| 67 | |
| 68 | self.say('\nBuilding net ...') |
| 69 | start = time.time() |
| 70 | self.graph = tf.Graph() |
| 71 | device_name = FLAGS.gpuName \ |
| 72 | if FLAGS.gpu > 0.0 else None |
| 73 | with tf.device(device_name): |
| 74 | with self.graph.as_default() as g: |
| 75 | self.build_forward() |
| 76 | self.setup_meta_ops() |
| 77 | self.say('Finished in {}s\n'.format( |
| 78 | time.time() - start)) |
| 79 | |
| 80 | def build_from_pb(self): |
| 81 | with tf.gfile.FastGFile(self.FLAGS.pbLoad, "rb") as f: |
nothing calls this directly
no test coverage detected