diff --git a/evaluate.ipynb b/evaluate.ipynb index f8db250..08e39df 100644 --- a/evaluate.ipynb +++ b/evaluate.ipynb @@ -6,7 +6,7 @@ "source": [ "## Range-Agnostic Multi-View Depth Estimation With Keyframe Selection\n", "\n", - "This notebooks provides a code example to use the pretrained models and reproduce the paper results." + "This notebook provides a code example to use the pretrained models and reproduce the paper results." ] }, { @@ -20,7 +20,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ @@ -33,10 +33,11 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ + "from dotenv import load_dotenv\n", "from lib.dataset import (\n", " BlendedMVSDataModule,\n", " TartanairDataModule,\n", @@ -51,12 +52,13 @@ "from ramdepth import Model\n", "from tqdm import tqdm\n", "import warnings\n", - "warnings.filterwarnings(\"ignore\", category=UserWarning)" + "warnings.filterwarnings(\"ignore\", category=UserWarning)\n", + "load_dotenv();" ] }, { "cell_type": "code", - "execution_count": 12, + "execution_count": null, "metadata": {}, "outputs": [], "source": [ @@ -87,8 +89,11 @@ " \"unrealstereo4k\": \"blended\",\n", " \"tartanair\": \"tartanair\",\n", " }[dataset],\n", - " normalize_pose=dataset != \"unrealstereo4k\"\n", - " and unreal_benchmark != \"stereo\",\n", + " normalize_pose=not (\n", + " dataset == \"unrealstereo4k\"\n", + " and unreal_benchmark == \"stereo\"\n", + " or dataset == \"tartanair\"\n", + " ),\n", " device=device,\n", ")" ] @@ -104,46 +109,9 @@ }, { "cell_type": "code", - "execution_count": 13, + "execution_count": null, "metadata": {}, - "outputs": [ - { - "name": "stderr", - "output_type": "stream", - "text": [ - " 0%| | 0/915 [00:00" - ] - }, - "metadata": {}, - "output_type": "display_data" - } - ], + "outputs": [], "source": [ "dl_iter = iter(dl)\n", "for i in range(sample_idx):\n",