Skip to content
Snippets Groups Projects

Shape dimensions

Merged Naser Azizi requested to merge shape_dimensions into main
Files
2
import robokudo.types.core
import robokudo.types.tf
import numpy as np
class Classification(robokudo.types.core.Annotation):
def __init__(self):
@@ -26,6 +26,26 @@ class Plane(robokudo.types.core.Annotation):
self.inliers = [] # List of pointcloud indices
class Shape(robokudo.types.core.Annotation):
def __init__(self):
super().__init__()
self.inliers = [] # List of pointcloud indices
self.type = '' # Can be Cuboid, Sphere, etc.
class Cuboid(Shape):
def __init__(self):
super().__init__()
self.model = [] # Has shape (3,4) - 3 plane equations
class Sphere(Shape):
def __init__(self):
super().__init__()
self.radius = 0.0 # Radius of sphere shape
self.center = np.zeros(3) # ndarray with shape (3,) represents a single coordinate
# Usually a 2D histogram containing hue and saturation
class ColorHistogram(robokudo.types.core.Annotation):
def __init__(self):
Loading