def _init_inception()

in inception.py [0:0]

• 37 lines of code
• 9 McCabe index (conditional complexity)

def _init_inception():
  global softmax
  if not os.path.exists(MODEL_DIR):
    os.makedirs(MODEL_DIR)
  filename = DATA_URL.split('/')[-1]
  filepath = os.path.join(MODEL_DIR, filename)
  if not os.path.exists(filepath):
    def _progress(count, block_size, total_size):
      sys.stdout.write('\r>> Downloading %s %.1f%%' % (
          filename, float(count * block_size) / float(total_size) * 100.0))
      sys.stdout.flush()
    filepath, _ = urllib.request.urlretrieve(DATA_URL, filepath, _progress)
    print()
    statinfo = os.stat(filepath)
    print('Succesfully downloaded', filename, statinfo.st_size, 'bytes.')
  tarfile.open(filepath, 'r:gz').extractall(MODEL_DIR)
  with tf.gfile.FastGFile(os.path.join(
      MODEL_DIR, 'classify_image_graph_def.pb'), 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    _ = tf.import_graph_def(graph_def, name='')
  # Works with an arbitrary minibatch size.
  pool3 = sess.graph.get_tensor_by_name('pool_3:0')
  ops = pool3.graph.get_operations()
  for op_idx, op in enumerate(ops):
      for o in op.outputs:
          shape = o.get_shape()
          shape = [s.value for s in shape]
          new_shape = []
          for j, s in enumerate(shape):
              if s == 1 and j == 0:
                  new_shape.append(None)
              else:
                  new_shape.append(s)
          o.set_shape(tf.TensorShape(new_shape))
  w = sess.graph.get_operation_by_name("softmax/logits/MatMul").inputs[1]
  logits = tf.matmul(tf.squeeze(pool3, [1, 2]), w)
  softmax = tf.nn.softmax(logits)