update

geom_solver.py

@@ -170,12 +170,12 @@ class GeomSolver(object):
 		self.vertices = centers
 		nvert = centers.shape[0]
 		# desired_vertices = (self.xyz[:,-1] > z_th).sum() // 300
-		desired_vertices = 2*nvert 
+		desired_vertices = int(2*nvert)
 		# desired_vertices = self.mean_vertices
 		if desired_vertices < self.min_vertices:
-			desired_vertices = self.mean_vertices
+			desired_vertices = self.min_vertices
 		if desired_vertices > self.max_vertices:
-			desired_vertices = self.mean_vertices
+			desired_vertices = self.max_vertices
 		# if self.broken_cams.any():
 		# 	vertices = centers
 		# 	print("There are broken cams.")
@@ -235,7 +235,7 @@ class GeomSolver(object):
 
 		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 : [1, 7], 1 : [3, 25], 2: [3, 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": 200,
+   "execution_count": null,
    "id": "88f4fc8f-efa9-404b-9073-c7d4a73f9075",
    "metadata": {},
    "outputs": [
@@ -184,38 +184,29 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "2.4183324229440974 2.038011601786833\n",
-      "1.9395643500589714 2.0791329825966307\n",
-      "2.815541573372287 1.9843719666126507\n",
-      "2.3254810143936755 1.524222149396748\n",
-      "2.3286533191380765 1.9072141124210245\n",
-      "2.342868026940067 1.9669375301897176\n",
+      "2.4183324229440974 1.9299500599980244\n",
+      "1.9395643500589714 2.0236575080719104\n",
+      "2.815541573372287 2.025451371291628\n",
+      "2.3254810143936755 1.6480476525775554\n",
+      "2.3286533191380765 1.907214134613452\n",
+      "2.342868026940067 1.9669375866465235\n",
       "2.04873204164575 1.8261703137595002\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.5473207425518596\n",
-      "2.135748984421359 1.677347152475311\n",
-      "2.3955711940075584 2.554900560046454\n",
-      "2.516982526006783 2.2057496776424905\n",
+      "2.167980973024546 1.564003268847725\n",
+      "2.571328055702413 2.0208250552562745\n",
+      "2.0127597511603774 2.0591274116795564\n",
+      "1.961080548873729 1.8200081714452052\n",
+      "2.1617889927813057 1.6349180723685486\n",
+      "2.0817899095268078 1.5980454240474442\n",
+      "2.484359575186481 1.7471060400666858\n",
+      "2.2997637652539 1.7949245589214717\n",
+      "2.440276048704617 2.547320743171852\n",
+      "2.135748984421359 1.6773471490746195\n",
+      "2.3955711940075584 2.5549005938032896\n",
+      "2.516982526006783 2.2057496863750923\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.8787443502537793\n"
+      "2.932654420428436 1.888064313171201\n",
+      "1.9387187366052139 1.6981983372339668\n"
      ]
     }
    ],
@@ -275,17 +266,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 185,
+   "execution_count": 210,
    "id": "a34d0a2b-9986-47cc-8a3f-c15397370c4d",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "0"
+       "29"
       ]
      },
-     "execution_count": 185,
+     "execution_count": 210,
      "metadata": {},
      "output_type": "execute_result"
     }