update

geom_solver.py

@@ -234,8 +234,8 @@ class GeomSolver(object):
 		    uvs.append(uv)
 
 		edges = []
-		thresholds_min_mean = {0 : [5, 7], 1 : [9, 25], 2: [30, 1000]}
-		# thresholds_min_mean = {0 : [1, 7], 1 : [1, 25], 2: [1, 1000]}
+		# thresholds_min_mean = {0 : [5, 7], 1 : [9, 25], 2: [30, 1000]}
+		thresholds_min_mean = {0 : [1, 7], 1 : [1, 25], 2: [1, 1000]}
 		for i in range(pyt_centers.shape[0]):
 			for j in range(i+1, pyt_centers.shape[0]):
 				etype = (self.is_apex[i] + self.is_apex[j])

testing.ipynb

@@ -176,7 +176,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 195,
+   "execution_count": 200,
    "id": "88f4fc8f-efa9-404b-9073-c7d4a73f9075",
    "metadata": {},
    "outputs": [
@@ -184,38 +184,38 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "2.4183324229440974 2.038011551581092\n",
+      "2.4183324229440974 2.038011601786833\n",
       "1.9395643500589714 2.0791329825966307\n",
-      "2.815541573372287 2.5366888251094535\n",
-      "2.3254810143936755 1.648047653136293\n",
-      "2.3286533191380765 1.9072141098956248\n",
-      "2.342868026940067 1.9669375395419237\n",
+      "2.815541573372287 1.9843719666126507\n",
+      "2.3254810143936755 1.524222149396748\n",
+      "2.3286533191380765 1.9072141124210245\n",
+      "2.342868026940067 1.9669375301897176\n",
       "2.04873204164575 1.8261703137595002\n",
-      "2.167980973024546 1.5160824709801253\n",
-      "2.571328055702413 2.020824999345514\n",
-      "2.0127597511603774 2.0591274017651693\n",
-      "1.961080548873729 1.8200081675273143\n",
-      "2.1617889927813057 1.6349180819215263\n",
-      "2.0817899095268078 1.5980454240474442\n",
-      "2.484359575186481 1.9029737257539197\n",
+      "2.167980973024546 1.5057396333492779\n",
+      "2.571328055702413 1.9803316090857763\n",
+      "2.0127597511603774 2.0591274587178017\n",
+      "1.961080548873729 1.9495190701464047\n",
+      "2.1617889927813057 1.6349180906181875\n",
+      "2.0817899095268078 1.5980454220244267\n",
+      "2.484359575186481 1.8898610018691997\n",
       "2.2997637652539 1.794924573842486\n",
-      "2.440276048704617 2.5473207175534065\n",
-      "2.135748984421359 1.6773471477819646\n",
-      "2.3955711940075584 2.5549005605155117\n",
-      "2.516982526006783 2.2057496708814113\n",
-      "2.521576698581939 1.615795718071218\n",
-      "1.8495344082304994 1.57835603284867\n",
-      "2.932654420428436 1.8880643234492105\n",
-      "1.9387187366052139 1.6981983307675732\n",
-      "2.135296338517323 1.8054484663488366\n",
-      "2.527322316920579 1.7985373132549314\n",
-      "2.681606928809334 2.2261780476238493\n",
-      "2.4613439446210306 1.6635009702924486\n",
-      "2.3497006297365304 1.815488520909667\n",
-      "2.2946897561795074 1.6075314409425536\n",
+      "2.440276048704617 2.5473207425518596\n",
+      "2.135748984421359 1.677347152475311\n",
+      "2.3955711940075584 2.554900560046454\n",
+      "2.516982526006783 2.2057496776424905\n",
+      "2.521576698581939 1.6157957679817523\n",
+      "1.8495344082304994 1.578356057609313\n",
+      "2.932654420428436 1.8880643428184267\n",
+      "1.9387187366052139 1.6981983296104317\n",
+      "2.135296338517323 1.81075437070104\n",
+      "2.527322316920579 1.7985373080000897\n",
+      "2.681606928809334 2.226178041689217\n",
+      "2.4613439446210306 1.5399683591875555\n",
+      "2.3497006297365304 1.8154885619111523\n",
+      "2.2946897561795074 1.6075314334284299\n",
       "2.2292662656029454 2.059612285543195\n",
       "Averages\n",
-      "2.312343783912538 1.9030379122529486\n"
+      "2.312343783912538 1.8787443502537793\n"
      ]
     }
    ],
@@ -226,7 +226,7 @@
     "torch.manual_seed(0)\n",
     "# One shard of the dataset\n",
     "dataset = wds.WebDataset(hf_hub_download(repo_id='usm3d/hoho-train-set',\n",
-    "            filename='data/train/hoho_v3_001-of-032.tar.gz',\n",
+    "            filename='data/train/hoho_v3_000-of-032.tar.gz',\n",
     "            repo_type=\"dataset\"))\n",
     "\n",
     "dataset = dataset.decode()\n",