Fix errors in smurf code.

PiperOrigin-RevId: 401825392

smurf/apply_smurf.py

@@ -40,7 +40,6 @@
 except:  # pylint:disable=bare-except
   print('Missing cv2 dependency. Please install opencv-python.')
 
-from cvx2 import latest as cv2  # pylint:disable=g-import-not-at-top
 
 
 flags.DEFINE_string('data_dir', '', 'Directory with images to run on. Images '

smurf/smurf_evaluator.py

@@ -25,6 +25,32 @@
 import tensorflow as tf
 
 
+from smurf import smurf_data
+from smurf import smurf_flags
+from smurf import smurf_plotting
+from smurf import smurf_net
+
+FLAGS = flags.FLAGS
+
+
+def build_network(batch_size):
+  """Builds the model architecture."""
+  return smurf_net.SMURFNet(
+      checkpoint_dir=FLAGS.checkpoint_dir,
+      optimizer=FLAGS.optimizer,
+      dropout_rate=0.,
+      feature_architecture=FLAGS.feature_architecture,
+      flow_architecture=FLAGS.flow_architecture,
+      size=(batch_size, FLAGS.height, FLAGS.width),
+      occlusion_estimation=FLAGS.occlusion_estimation,
+      smoothness_at_level=FLAGS.smoothness_at_level,
+      use_float16=True,
+  )
+
+
+def evaluate():
+  """Eval happens on GPU or CPU, and evals each checkpoint as it appears."""
+
   candidate_checkpoint = None
   smurf =build_network(batch_size=1)