Unverified Commit d90d5280 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #3206 from cclauss/from-six.moves-import-xrange

from six.moves import xrange (en masse)
parents 6fc65ee6 848cc592
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
"""Read image sequence.""" """Read image sequence."""
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
......
...@@ -21,6 +21,7 @@ import sys ...@@ -21,6 +21,7 @@ import sys
import numpy as np import numpy as np
import scipy.misc import scipy.misc
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
......
...@@ -18,6 +18,7 @@ from __future__ import absolute_import ...@@ -18,6 +18,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
from object_detection.core import standard_fields from object_detection.core import standard_fields
......
...@@ -21,6 +21,7 @@ Example box operations that are supported: ...@@ -21,6 +21,7 @@ Example box operations that are supported:
""" """
import numpy as np import numpy as np
from six.moves import xrange
from object_detection.utils import np_box_list from object_detection.utils import np_box_list
from object_detection.utils import np_box_ops from object_detection.utils import np_box_ops
......
...@@ -20,6 +20,7 @@ In some cases this is just an additional linear layer on the policy. ...@@ -20,6 +20,7 @@ In some cases this is just an additional linear layer on the policy.
In other cases, it is a completely separate neural network. In other cases, it is a completely separate neural network.
""" """
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np
......
...@@ -20,6 +20,7 @@ from __future__ import absolute_import ...@@ -20,6 +20,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np
import pickle import pickle
......
...@@ -20,6 +20,7 @@ from __future__ import division ...@@ -20,6 +20,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import numpy as np import numpy as np
from six.moves import xrange
class spaces(object): class spaces(object):
......
...@@ -22,6 +22,7 @@ import tensorflow as tf ...@@ -22,6 +22,7 @@ import tensorflow as tf
import random import random
import os import os
import numpy as np import numpy as np
from six.moves import xrange
import pickle import pickle
gfile = tf.gfile gfile = tf.gfile
......
...@@ -22,6 +22,7 @@ import gym ...@@ -22,6 +22,7 @@ import gym
import numpy as np import numpy as np
import random import random
from six.moves import xrange
import env_spec import env_spec
......
...@@ -25,6 +25,7 @@ from __future__ import absolute_import ...@@ -25,6 +25,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np
import scipy.optimize import scipy.optimize
......
...@@ -20,6 +20,7 @@ Implements replay buffer in Python. ...@@ -20,6 +20,7 @@ Implements replay buffer in Python.
import random import random
import numpy as np import numpy as np
from six.moves import xrange
class ReplayBuffer(object): class ReplayBuffer(object):
......
...@@ -25,6 +25,7 @@ import random ...@@ -25,6 +25,7 @@ import random
import os import os
import pickle import pickle
from six.moves import xrange
import controller import controller
import model import model
import policy import policy
......
...@@ -24,6 +24,7 @@ from __future__ import absolute_import ...@@ -24,6 +24,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np
......
...@@ -19,6 +19,7 @@ from __future__ import absolute_import ...@@ -19,6 +19,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
slim = tf.contrib.slim slim = tf.contrib.slim
...@@ -108,5 +109,3 @@ def add_volume_iou_metrics(inputs, outputs): ...@@ -108,5 +109,3 @@ def add_volume_iou_metrics(inputs, outputs):
names_to_values['volume_iou'] = tmp_values * 3.0 names_to_values['volume_iou'] = tmp_values * 3.0
names_to_updates['volume_iou'] = tmp_updates names_to_updates['volume_iou'] = tmp_updates
return names_to_values, names_to_updates return names_to_values, names_to_updates
...@@ -21,6 +21,7 @@ from __future__ import print_function ...@@ -21,6 +21,7 @@ from __future__ import print_function
import os import os
import numpy as np import numpy as np
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import losses import losses
......
...@@ -21,6 +21,7 @@ from __future__ import print_function ...@@ -21,6 +21,7 @@ from __future__ import print_function
import os import os
import numpy as np import numpy as np
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import input_generator import input_generator
...@@ -191,19 +192,19 @@ def get_train_op_for_scope(loss, optimizer, scopes, params): ...@@ -191,19 +192,19 @@ def get_train_op_for_scope(loss, optimizer, scopes, params):
def get_metrics(inputs, outputs, params): def get_metrics(inputs, outputs, params):
"""Aggregate the metrics for rotator model. """Aggregate the metrics for rotator model.
Args: Args:
inputs: Input dictionary of the rotator model. inputs: Input dictionary of the rotator model.
outputs: Output dictionary returned by the rotator model. outputs: Output dictionary returned by the rotator model.
params: Hyperparameters of the rotator model. params: Hyperparameters of the rotator model.
Returns: Returns:
names_to_values: metrics->values (dict). names_to_values: metrics->values (dict).
names_to_updates: metrics->ops (dict). names_to_updates: metrics->ops (dict).
""" """
names_to_values = dict() names_to_values = dict()
names_to_updates = dict() names_to_updates = dict()
tmp_values, tmp_updates = metrics.add_image_pred_metrics( tmp_values, tmp_updates = metrics.add_image_pred_metrics(
inputs, outputs, params.num_views, 3*params.image_size**2) inputs, outputs, params.num_views, 3*params.image_size**2)
names_to_values.update(tmp_values) names_to_values.update(tmp_values)
...@@ -217,7 +218,7 @@ def get_metrics(inputs, outputs, params): ...@@ -217,7 +218,7 @@ def get_metrics(inputs, outputs, params):
for name, value in names_to_values.iteritems(): for name, value in names_to_values.iteritems():
slim.summaries.add_scalar_summary( slim.summaries.add_scalar_summary(
value, name, prefix='eval', print_summary=True) value, name, prefix='eval', print_summary=True)
return names_to_values, names_to_updates return names_to_values, names_to_updates
......
...@@ -22,6 +22,7 @@ import abc ...@@ -22,6 +22,7 @@ import abc
import os import os
import numpy as np import numpy as np
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
import input_generator import input_generator
......
...@@ -21,6 +21,7 @@ from __future__ import print_function ...@@ -21,6 +21,7 @@ from __future__ import print_function
import os import os
import numpy as np import numpy as np
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
from tensorflow import app from tensorflow import app
......
...@@ -28,6 +28,7 @@ from mpl_toolkits.mplot3d import axes3d as p3 # pylint:disable=unused-import ...@@ -28,6 +28,7 @@ from mpl_toolkits.mplot3d import axes3d as p3 # pylint:disable=unused-import
import numpy as np import numpy as np
from PIL import Image from PIL import Image
from skimage import measure from skimage import measure
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
...@@ -116,4 +117,3 @@ def visualize_voxel_scatter(points, vis_size=128): ...@@ -116,4 +117,3 @@ def visualize_voxel_scatter(points, vis_size=128):
vis_size, vis_size, 3) vis_size, vis_size, 3)
p.close('all') p.close('all')
return data return data
...@@ -19,6 +19,7 @@ r"""Utility functions for Real NVP. ...@@ -19,6 +19,7 @@ r"""Utility functions for Real NVP.
# pylint: disable=dangerous-default-value # pylint: disable=dangerous-default-value
import numpy import numpy
from six.moves import xrange
import tensorflow as tf import tensorflow as tf
from tensorflow.python.framework import ops from tensorflow.python.framework import ops
......
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