GraphNodes¶

class GraphNodes(nodes, sly_id=None, class_id=None, labeler_login=None, updated_at=None, created_at=None)[source]¶

Bases: Geometry

Geometry representing a graph of labeled keypoints: a collection of Node objects for pose estimation, skeletons, or structured point sets (e.g. nose, left_eye, right_shoulder).

GraphNodes geometry for a single Label. GraphNodes class object is immutable.

Parameters:

nodes : dict¶: Dict or List containing nodes of graph
sly_id : int, optional¶: GraphNodes ID in Supervisely server.
class_id : int, optional¶: ID of ObjClass to which GraphNodes belongs.
labeler_login : str, optional¶: Login of the user who created GraphNodes.
updated_at : str, optional¶: Date and Time when GraphNodes was modified last. Date Format: Year:Month:Day:Hour:Minute:Seconds. Example: ‘2021-01-22T19:37:50.158Z’.
created_at : str, optional¶: Date and Time when GraphNodes was created. Date Format is the same as in “updated_at” parameter.

Usage Example:

import supervisely as sly
from supervisely.geometry.graph import Node, GraphNodes

vertex_1 = Node(sly.PointLocation(5, 5))
vertex_2 = Node(sly.PointLocation(100, 100))
vertex_3 = Node(sly.PointLocation(200, 250))
nodes = {0: vertex_1, 1: vertex_2, 2: vertex_3}
figure = GraphNodes(nodes)

Methods

`allowed_transforms`	Returns the allowed transforms for the GraphNodes.
`clone`	Makes a copy of the GraphNodes.
`config_from_json`	Convert graph template from json format
`config_to_json`	Convert graph template to json format
`convert`	Convert geometry to another geometry shape.
`crop`	Crops current GraphNodes.
`draw`
`draw_contour`	Draws the figure contour on a given bitmap canvas :param bitmap: np.ndarray :param color: [R, G, B] :param thickness: (int) :param config: drawing config specific to a concrete subclass, e.g. per edge colors.
`fliplr`	Flips current GraphNodes in horizontal.
`flipud`	Flips current GraphNodes in vertical.
`from_json`	Convert a json dict to GraphNodes.
`geometry_name`
`get_mask`	Returns 2D boolean mask of the geometry.
`name`	Get the name of the geometry.
`relative_crop`	Crops current GraphNodes with given rectangle and shifts it on value of rectangle left top angle.
`resize`	Resizes current GraphNodes.
`rotate`	Rotates current GraphNodes.
`scale`	Scales current GraphNodes.
`to_bbox`	Create Rectangle object from current GraphNodes.
`to_json`	Convert the GraphNodes to list.
`transform`
`transform_locations`
`translate`	Translates current GraphNodes.
`validate`	Checks the graph for correctness and compliance with the template

Attributes

`area`	GraphNodes area.
`items_json_field`
`nodes`	Copy of GraphNodes nodes.

classmethod allowed_transforms()[source]¶: Returns the allowed transforms for the GraphNodes.

classmethod from_json(data)[source]¶

Convert a json dict to GraphNodes. Read more about Supervisely format.

Parameters:

data : dict¶: GraphNodes in json format as a dict.

Returns:

GraphNodes from json.

Return type:

GraphNodes

Usage Example:

figure_json = {
    "nodes": {
        "0": {"loc": [5, 5]},
        "1": {"loc": [100, 100]},
        "2": {"loc": [250, 200]}
    }
}
from supervisely.geometry.graph import GraphNodes
figure = GraphNodes.from_json(figure_json)

classmethod name()¶

Get the name of the geometry.

Same as geometry_name(), but shorter. In order to make the code more concise.

Returns:: string with name of geometry

static config_from_json(config)[source]¶

Convert graph template from json format

Parameters:

config : dict¶: dictionary(graph template) in json format

Returns:

dictionary(graph template) in json format

Return type:

dict

static config_to_json(config)[source]¶

Convert graph template to json format

Parameters:

config : dict¶: dictionary(graph template)

Returns:

dictionary(graph template) in json format

Return type:

dict

clone()[source]¶

Makes a copy of the GraphNodes.

Returns:

Copy of GraphNodes.

Return type:

GraphNodes

Usage Example:

# Remember that GraphNodes class object is immutable, and we need to assign new instance of PointLocation to a new variable
new_figure = figure.clone()

convert(new_geometry, contour_radius=0, approx_epsilon=None)¶

Convert geometry to another geometry shape.

Parameters:

new_geometry¶: New geometry shape.
contour_radius : int¶: Radius of the contour.
approx_epsilon : float¶: Approximation epsilon.

Returns:

List of geometries.

Return type:

List[Geometry]

crop(rect)[source]¶

Crops current GraphNodes.

Parameters:

rect¶: Rectangle for crop.

Returns:

List of GraphNodes after crop.

Return type:

List[GraphNodes]

Usage Example:

import supervisely as sly

crop_figures = figure.crop(sly.Rectangle(0, 0, 300, 350))

draw(bitmap, color, thickness=1, config=None)¶

Parameters:

bitmap¶: np.ndarray
color¶: Color [R, G, B]
thickness=1¶: used only in Polyline and Point
config : dict¶: Drawing config specific to a concrete subclass, e.g. per edge colors

Raises:

NotImplementedError – if method is not implemented in subclass

draw_contour(bitmap, color, thickness=1, config=None)¶: Draws the figure contour on a given bitmap canvas :param bitmap: np.ndarray :param color: [R, G, B] :param thickness: (int) :param config: drawing config specific to a concrete subclass, e.g. per edge colors

fliplr(img_size)[source]¶

Flips current GraphNodes in horizontal.

Parameters:

img_size : Tuple[int, int]¶: Input image size (height, width) to which belongs GraphNodes.

Returns:

GraphNodes after flip in horizontal.

Return type:

GraphNodes

Usage Example:

# Remember that GraphNodes class object is immutable, and we need to assign new instance of Rectangle to a new variable
height, width = 300, 400
fliplr_figure = figure.fliplr((height, width))

flipud(img_size)[source]¶

Flips current GraphNodes in vertical.

Parameters:

img_size : Tuple[int, int]¶: Input image size (height, width).

Returns:

GraphNodes after flip in vertical.

Return type:

GraphNodes

Usage Example:

# Remember that GraphNodes class object is immutable, and we need to assign new instance of Rectangle to a new variable
height, width = 300, 400
flipud_figure = figure.flipud((height, width))

get_mask(img_size)¶

Returns 2D boolean mask of the geometry. With shape as img_size (height, width) and filled with True values inside the geometry and False values outside. dtype = np.bool shape = img_size

Parameters:

img_size : Tuple[int, int]¶: size of the image (height, width)

Returns:

2D boolean mask of the geometry

Return type:

np.ndarray

relative_crop(rect)[source]¶

Crops current GraphNodes with given rectangle and shifts it on value of rectangle left top angle.

Parameters:

rect¶: Rectangle for crop.

Returns:

List of GraphNodes after relative crop.

Return type:

List[GraphNodes]

Usage Example:

import supervisely as sly

rel_crop_figures = figure.relative_crop(sly.Rectangle(0, 0, 300, 350))

resize(in_size, out_size)[source]¶

Resizes current GraphNodes.

Parameters:

in_size : Tuple[int, int]¶: Input image size (height, width) to which belongs GraphNodes.
out_size : Tuple[int, int]¶: Desired output image size (height, width) to which belongs GraphNodes.

Returns:

GraphNodes after resize.

Return type:

GraphNodes

Usage Example:

# Remember that GraphNodes class object is immutable, and we need to assign new instance of Rectangle to a new variable
in_height, in_width = 300, 400
out_height, out_width = 600, 800
resize_figure = figure.resize((in_height, in_width), (out_height, out_width))

rotate(rotator)[source]¶

Rotates current GraphNodes.

Parameters:

rotator¶: Class for image rotation.

Returns:

GraphNodes after rotation.

Return type:

GraphNodes

Usage Example:

from supervisely.geometry.image_rotator import ImageRotator

# Remember that GraphNodes class object is immutable, and we need to assign new instance of Rectangle to a new variable
height, width = 300, 400
rotator = ImageRotator((height, width), 25)
rotate_figure = figure.rotate(rotator)

scale(factor)[source]¶

Scales current GraphNodes.

Parameters:

factor : float¶: Scale parameter.

Returns:

GraphNodes object

Return type:

GraphNodes

Usage Example:

# Remember that GraphNodes class object is immutable, and we need to assign new instance of Rectangle to a new variable
scale_figure = figure.scale(0.75)

to_bbox()[source]¶

Create Rectangle object from current GraphNodes.

Returns:

Rectangle from GraphNodes.

Return type:

Rectangle

Usage Example:

rectangle = figure.to_bbox()

to_json()[source]¶

Convert the GraphNodes to list. Read more about Supervisely format.

Returns:

Json format as a dict

Return type:

dict

Usage Example:

import supervisely as sly
from supervisely.geometry.graph import Node, GraphNodes

vertex_1 = Node(sly.PointLocation(5, 5))
vertex_2 = Node(sly.PointLocation(100, 100))
vertex_3 = Node(sly.PointLocation(200, 250))
nodes = {0: vertex_1, 1: vertex_2, 2: vertex_3}
figure = GraphNodes(nodes)

figure_json = figure.to_json()
print(figure_json)
# Output: {
#    "nodes": {
#        "0": {
#            "loc": [5, 5]
#        },
#        "1": {
#            "loc": [100, 100]
#        },
#        "2": {
#            "loc": [250, 200]
#        }
#    }
# }

transform(transform_fn)[source]¶

Parameters:

transform_fn : function¶: Function to convert GraphNodes.

Returns:

GraphNodes after transformation.

Return type:

GraphNodes

transform_locations(transform_fn)[source]¶

Parameters:

transform_fn : function¶: Function to convert GraphNodes location.

Returns:

GraphNodes after transformation.

Return type:

GraphNodes

translate(drow, dcol)[source]¶

Translates current GraphNodes.

Parameters:

drow : int¶: Horizontal shift.
dcol : int¶: Vertical shift.

Returns:

GraphNodes after translate.

Return type:

GraphNodes

Usage Example:

# Remember that GraphNodes class object is immutable, and we need to assign new instance of Rectangle to a new variable
translate_figure = figure.translate(150, 250)

validate(name, settings)[source]¶

Checks the graph for correctness and compliance with the template

Parameters:

name : str¶: Name of the geometry.
settings : dict¶: Settings of the geometry.

Raises:

ValueError – if graph contains nodes not declared in the template

property area : float¶

GraphNodes area.

Returns:

Area of current GraphNodes, always 0.0

Return type:

float

Usage Example:

print(figure.area)
# Output: 0.0

property nodes : dict[str, dict]¶

Copy of GraphNodes nodes.

Returns:: GraphNodes nodes
Return type:: dict