Пример #1
0
    def upload_catalog(self, catalog: Catalog) -> None:
        """Upload a catalog to the draft.

        Arguments:
            catalog: :class:`~tensorbay.label.catalog.Catalog` to upload.

        """
        self._status.check_authority_for_draft()

        put_data: Dict[str, Any] = {"catalog": catalog.dumps()}
        put_data.update(self._status.get_status_info())

        self._client.open_api_do("PUT",
                                 "labels/catalogs",
                                 self._dataset_id,
                                 json=put_data)
Пример #2
0
    def test_dumps(self, categories, keypoints, catalog_contents):
        catalog = Catalog()
        catalog.box2d = Box2DSubcatalog()
        catalog.box3d = Box3DSubcatalog()
        catalog.polygon = PolygonSubcatalog()
        catalog.polyline2d = Polyline2DSubcatalog()
        catalog.sentence = SentenceSubcatalog()

        classificationsubcatalog = ClassificationSubcatalog()
        classificationsubcatalog.categories = categories
        catalog.classification = classificationsubcatalog

        keypoints2dsubcatalog = Keypoints2DSubcatalog()
        keypoints2dsubcatalog._keypoints = [keypoints]
        catalog.keypoints2d = keypoints2dsubcatalog

        assert catalog.dumps() == catalog_contents