mirror of
https://github.com/opencv/opencv.git
synced 2026-07-26 13:53:03 +04:00
Alexander Smorkalov
fc3803c67b
Dedicated pointcloud module #29224 OpenCV contrib: https://github.com/opencv/opencv_contrib/pull/4134 ### Pull Request Readiness Checklist See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [x] I agree to contribute to the project under Apache 2 License. - [x] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [ ] The PR is proposed to the proper branch - [ ] There is a reference to the original bug report and related work - [ ] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [ ] The feature is well documented and sample code can be built with the project CMake
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Python 2/3 compatibility
|
|
from __future__ import print_function
|
|
|
|
import os, numpy
|
|
import math
|
|
import unittest
|
|
import cv2 as cv
|
|
|
|
from tests_common import NewOpenCVTests
|
|
|
|
class raster_test(NewOpenCVTests):
|
|
|
|
def test_loadCloud(self):
|
|
fin = self.find_file("pointcloudio/orig.obj")
|
|
points, normals, colors = cv.loadPointCloud(fin)
|
|
|
|
if points.shape != (8, 1, 3):
|
|
self.fail('point array should be 8x1x3')
|
|
if normals.shape != (6, 1, 3):
|
|
self.fail('normals array should be 6x1x3')
|
|
if colors.shape != (8, 1, 3):
|
|
self.fail('colors array should be 8x1x3')
|
|
|
|
def test_loadMesh(self):
|
|
fin = self.find_file("pointcloudio/orig.obj")
|
|
points, indices, normals, colors, texCoords = cv.loadMesh(fin)
|
|
goodShapes = [(1, 18, 3), (18, 1, 3)]
|
|
errorMsg = "%s array should be 18x1x3 or 1x18x3"
|
|
for a, s in [(points, 'points'), (normals, 'normals'), (colors, 'colors')]:
|
|
if a.shape not in goodShapes:
|
|
self.fail(errorMsg % s)
|
|
|
|
if texCoords.shape not in [(1, 18, 2), (18, 1, 2), (18, 2)]:
|
|
self.fail('texture coordinates array should be 1x18x2 or 18x1x2')
|
|
if isinstance(indices, numpy.ndarray):
|
|
if indices.shape not in [(1, 6, 3), (6, 1, 3)]:
|
|
self.fail('indices array should be 1x6x3 or 6x1x3')
|
|
elif isinstance(indices, list) or isinstance(indices, tuple):
|
|
for i in indices:
|
|
if len(indices) != 6 or i.shape != (1, 3):
|
|
self.fail('indices array should be 1x6x3 or 6x1x3')
|
|
|
|
if __name__ == '__main__':
|
|
NewOpenCVTests.bootstrap()
|