diff --git a/scripts/summarize.ipynb b/scripts/summarize.ipynb index 697c9ca..b11c6a2 100644 --- a/scripts/summarize.ipynb +++ b/scripts/summarize.ipynb @@ -23,8 +23,7 @@ "outputs": [], "source": [ "import os\n", - "import numpy as np\n", - "from google3.pyglib import gfile" + "import numpy as np" ] }, { @@ -48,7 +47,7 @@ " values = []\n", " for metric_name in metric_names:\n", " filename = os.path.join(folder, scene_name, 'test_preds', f'{metric_name}_{num_iters}.txt')\n", - " with gfile.Open(filename) as f:\n", + " with open(filename) as f:\n", " v = np.array([float(s) for s in f.readline().split(' ')])\n", " values.append(np.mean(np.reshape(v, [-1, num_buckets]), 0))\n", " results.append(np.concatenate(values))\n",