Skip to content

Commit

Permalink
Merge pull request #172 from oarriaga/pix2pose_merge
Browse files Browse the repository at this point in the history 
PIX2POSE merge
  • Loading branch information
@oarriaga
oarriaga authored Feb 11, 2022
2 parents d345165 + 5377784 commit 67e2fcd
Show file tree
Hide file tree
Showing 53 changed files with 2,304 additions and 2,145 deletions.
132 changes: 94 additions & 38 deletions docs/structure.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@
from paz.backend import camera
from paz.backend import render
from paz.backend import keypoints
from paz.backend import quaternion
from paz.backend import groups
from paz.backend import image
from paz.backend.image import draw
from paz.backend.image import opencv_image
from paz.abstract import messages
from paz.abstract import processor
from paz.abstract import loader
Expand Down Expand Up @@ -41,17 +41,23 @@
boxes.to_center_form,
boxes.to_one_hot,
boxes.to_normalized_coordinates,
boxes.to_corner_form
boxes.to_corner_form,
boxes.extract_bounding_box_corners
],
},


{
'page': 'backend/keypoints.md',
'functions': [
keypoints.build_cube_points3D,
keypoints.normalize_keypoints2D,
keypoints.denormalize_keypoints2D,
keypoints.project_to_image,
keypoints.solve_PnP_RANSAC,
keypoints.arguments_to_image_points2D,
keypoints.points3D_to_RGB,
keypoints.cascade_classifier,
keypoints.denormalize_keypoints,
keypoints.normalize_keypoints,
keypoints.project_points3D,
keypoints.solve_PNP,
keypoints.translate_keypoints
Expand All @@ -60,9 +66,18 @@


{
'page': 'backend/quaternion.md',
'page': 'backend/groups.md',
'functions': [
quaternion.rotation_vector_to_quaternion
groups.rotation_vector_to_quaternion,
groups.homogenous_quaternion_to_rotation_matrix,
groups.quaternion_to_rotation_matrix,
groups.to_affine_matrix,
groups.rotation_vector_to_rotation_matrix,
groups.build_rotation_matrix_x,
groups.build_rotation_matrix_y,
groups.build_rotation_matrix_z,
groups.compute_norm_SO3,
groups.calculate_canonical_rotation
],
},

Expand Down Expand Up @@ -92,37 +107,44 @@
draw.draw_rectangle,
draw.lincolor,
draw.put_text,
draw.make_mosaic
draw.make_mosaic,
draw.draw_points2D
],
},


{
'page': 'backend/image.md',
'functions': [
opencv_image.cast_image,
opencv_image.resize_image,
opencv_image.convert_color_space,
opencv_image.load_image,
opencv_image.random_saturation,
opencv_image.random_brightness,
opencv_image.random_contrast,
opencv_image.random_hue,
opencv_image.random_flip_left_right,
opencv_image.show_image,
opencv_image.warp_affine,
opencv_image.write_image,
opencv_image.random_shape_crop,
opencv_image.make_random_plain_image,
opencv_image.blend_alpha_channel,
opencv_image.concatenate_alpha_mask,
opencv_image.split_and_normalize_alpha_channel,
opencv_image.gaussian_image_blur,
opencv_image.median_image_blur,
opencv_image.random_image_blur,
opencv_image.translate_image,
opencv_image.sample_scaled_translation,
opencv_image.get_rotation_matrix
image.resize_image,
image.convert_color_space,
image.load_image,
image.show_image,
image.warp_affine,
image.write_image,
image.gaussian_image_blur,
image.median_image_blur,
image.get_rotation_matrix,
image.cast_image,
image.random_saturation,
image.random_brightness,
image.random_contrast,
image.random_hue,
image.flip_left_right,
image.random_flip_left_right,
image.crop_image,
image.image_to_normalized_device_coordinates,
image.normalized_device_coordinates_to_image,
image.random_shape_crop,
image.make_random_plain_image,
image.blend_alpha_channel,
image.concatenate_alpha_mask,
image.split_and_normalize_alpha_channel,
image.random_image_blur,
image.translate_image,
image.sample_scaled_translation,
image.replace_lower_than_threshold,
image.normalize_min_max
],
},

Expand Down Expand Up @@ -228,7 +250,9 @@
losses.KeypointNetLoss,
losses.DiceLoss,
losses.FocalLoss,
losses.JaccardLoss
losses.JaccardLoss,
losses.WeightedReconstruction,
losses.WeightedReconstructionWithError
],
},

Expand Down Expand Up @@ -262,7 +286,12 @@
processors.ConcatenateAlphaMask,
processors.BlendRandomCroppedBackground,
processors.AddOcclusion,
processors.TranslateImage
processors.TranslateImage,
processors.ImageToNormalizedDeviceCoordinates,
processors.NormalizedDeviceCoordinatesToImage,
processors.ReplaceLowerThanThreshold,
processors.GetNonZeroValues,
processors.GetNonZeroArguments
]
},

Expand All @@ -273,7 +302,8 @@
processors.DrawBoxes2D,
processors.DrawKeypoints2D,
processors.DrawBoxes3D,
processors.DrawRandomPolygon
processors.DrawRandomPolygon,
processors.DrawPose6D
]
},

Expand All @@ -293,7 +323,6 @@
},



{
'page': 'processors/detection.md',
'classes': [
Expand Down Expand Up @@ -324,14 +353,18 @@
processors.PartitionKeypoints,
processors.ProjectKeypoints,
processors.RemoveKeypointsDepth,
processors.TranslateKeypoints
processors.TranslateKeypoints,
processors.DenormalizeKeypoints2D,
processors.NormalizeKeypoints2D,
processors.ArgumentsToImageKeypoints2D,
]
},

{
'page': 'processors/pose.md',
'classes': [
processors.SolvePNP
processors.SolvePNP,
processors.SolveChangingObjectPnPRANSAC
]
},

Expand All @@ -343,6 +376,16 @@
]
},

{
'page': 'processors/groups.md',
'classes': [
processors.ToAffineMatrix,
processors.RotationVectorToQuaternion,
processors.RotationVectorToRotationMatrix,
]
},



{
'page': 'processors/standard.md',
Expand All @@ -364,7 +407,9 @@
processors.Concatenate,
processors.SelectElement,
processors.StochasticProcessor,
processors.Stochastic
processors.Stochastic,
processors.UnwrapDictionary,
processors.Scale
]
},

Expand Down Expand Up @@ -411,6 +456,17 @@
]
},

{
'page': 'pipelines/masks.md',
'classes': [
pipelines.RGBMaskToImagePoints2D,
pipelines.RGBMaskToObjectPoints3D,
pipelines.PredictRGBMask,
pipelines.Pix2Points
]
},



{
'page': 'pipelines/renderer.md',
Expand Down
4 changes: 2 additions & 2 deletions docs/templates/installation.md
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
## Installation
PAZ has only **three** dependencies: [Tensorflow2.0](https://www.tensorflow.org/), [OpenCV](https://opencv.org/) and [NumPy](https://numpy.org/).

Clone the repository and run inside it:
To install PAZ with pypi run:

`pip install . --user`
`pip install pypaz --user`
64 changes: 0 additions & 64 deletions examples/pix2pose/abstract_test.py
View file

This file was deleted.

Loading

0 comments on commit 67e2fcd

Please sign in to comment.