def test_add_bbox_noinit(tmpdir):
tmpdir = Path(tmpdir)
projects_found = sa.search_projects(
PROJECT_NAME_NOINIT, return_metadata=True
)
for pr in projects_found:
sa.delete_project(pr)
project = sa.create_project(
PROJECT_NAME_NOINIT, PROJECT_DESCRIPTION, "Vector"
)
sa.upload_images_from_folder_to_project(
project, PATH_TO_SAMPLE_PROJECT, annotation_status="InProgress"
)
sa.create_annotation_classes_from_classes_json(
project, PATH_TO_SAMPLE_PROJECT / "classes" / "classes.json"
)
sa.create_annotation_class(project, "test_add", "#FF0000")
images = sa.search_images(project, "example_image_1")
image_name = images[0]
sa.add_annotation_bbox_to_image(
project, image_name, [10, 10, 500, 100], "test_add"
)
sa.add_annotation_polygon_to_image(
project, image_name, [100, 100, 500, 500, 200, 300], "test_add"
)
annotations_new = sa.get_image_annotations(project,
image_name)["annotation_json"]
assert len(annotations_new) == 2
export = sa.prepare_export(project, include_fuse=True)
sa.download_export(project, export, tmpdir)
assert len(list(Path(tmpdir).rglob("*.*"))) == 4
def test_add_bbox_noinit(tmpdir):
tmpdir = Path(tmpdir)
projects_found = sa.search_projects(PROJECT_NAME_NOINIT,
return_metadata=True)
for pr in projects_found:
sa.delete_project(pr)
project = sa.create_project(PROJECT_NAME_NOINIT, PROJECT_DESCRIPTION,
"Vector")
sa.upload_images_from_folder_to_project(project,
PATH_TO_SAMPLE_PROJECT,
annotation_status="InProgress")
sa.create_annotation_classes_from_classes_json(
project, PATH_TO_SAMPLE_PROJECT / "classes" / "classes.json")
sa.create_annotation_class(project, "test_add", "#FF0000")
images = sa.search_images(project, "example_image_1")
image_name = images[0]
sa.add_annotation_bbox_to_image(project, image_name, [10, 10, 500, 100],
"test_add")
sa.add_annotation_polygon_to_image(project, image_name,
[100, 100, 500, 500, 200, 300],
"test_add")
annotations_new = sa.get_image_annotations(project,
image_name)["annotation_json"]
assert len(annotations_new["instances"]) == 2
export = sa.prepare_export(project, include_fuse=True)
sa.download_export(project, export, tmpdir)
non_empty_annotations = 0
json_files = tmpdir.glob("*.json")
for json_file in json_files:
json_ann = json.load(open(json_file))
if "instances" in json_ann and len(json_ann["instances"]) > 0:
non_empty_annotations += 1
assert len(json_ann["instances"]) == 2
assert non_empty_annotations == 1
def test_add_bbox(tmpdir):
tmpdir = Path(tmpdir)
projects_found = sa.search_projects(PROJECT_NAME, return_metadata=True)
for pr in projects_found:
if pr["name"] == PROJECT_NAME:
sa.delete_project(pr)
project = sa.create_project(PROJECT_NAME, PROJECT_DESCRIPTION, "Vector")
sa.upload_images_from_folder_to_project(PROJECT_NAME,
PATH_TO_SAMPLE_PROJECT,
annotation_status="InProgress")
sa.create_annotation_classes_from_classes_json(
project, PATH_TO_SAMPLE_PROJECT / "classes" / "classes.json")
sa.create_annotation_class(
project, "test_add", "#FF0000",
[{
"name": "height",
"attributes": [{
"name": "tall"
}, {
"name": "short"
}]
}])
sa.upload_annotations_from_folder_to_project(project,
PATH_TO_SAMPLE_PROJECT)
images = sa.search_images(project, "example_image_1")
image_name = images[0]
annotations = sa.get_image_annotations(project,
image_name)["annotation_json"]
sa.add_annotation_bbox_to_image(project, image_name, [10, 10, 500, 100],
"test_add")
sa.add_annotation_polyline_to_image(project, image_name,
[110, 110, 510, 510, 600, 510],
"test_add")
sa.add_annotation_polygon_to_image(project, image_name,
[100, 100, 500, 500, 200, 300],
"test_add", [{
"name": "tall",
"groupName": "height"
}])
sa.add_annotation_point_to_image(project, image_name, [250, 250],
"test_add")
sa.add_annotation_ellipse_to_image(project, image_name,
[405, 405, 20, 70, 15], "test_add")
sa.add_annotation_template_to_image(project, image_name,
[600, 30, 630, 30, 615, 60],
[1, 3, 2, 3], "test_add")
sa.add_annotation_cuboid_to_image(project, image_name,
[800, 500, 900, 600, 850, 450, 950, 700],
"test_add")
sa.add_annotation_comment_to_image(project, image_name, "hey", [100, 100],
"*****@*****.**", True)
annotations_new = sa.get_image_annotations(project,
image_name)["annotation_json"]
json.dump(annotations_new, open(tmpdir / "new_anns.json", "w"))
assert len(annotations_new["instances"]) + len(
annotations_new["comments"]) == len(annotations["instances"]) + len(
annotations["comments"]) + 8
export = sa.prepare_export(project, include_fuse=True)
sa.download_export(project, export, tmpdir)
df = sa.aggregate_annotations_as_df(tmpdir)
print(df)
print(image_name)
num = len(
df[df["imageName"] == image_name]["instanceId"].dropna().unique())
assert num == len(
annotations["instances"]
) - 3 + 7 # -6 for 3 comments and 3 invalid annotations, className or attributes
def test_fuse_image_create_vector(tmpdir):
tmpdir = Path(tmpdir)
projects = sa.search_projects(PROJECT_NAME_VECTOR, return_metadata=True)
for project in projects:
sa.delete_project(project)
project = sa.create_project(PROJECT_NAME_VECTOR, "test", "Vector")
sa.upload_image_to_project(
project,
"./tests/sample_project_vector/example_image_1.jpg",
annotation_status="QualityCheck")
sa.create_annotation_classes_from_classes_json(
project, "./tests/sample_project_vector/classes/classes.json")
sa.add_annotation_bbox_to_image(project, "example_image_1.jpg",
[20, 20, 40, 40], "Human")
sa.add_annotation_polygon_to_image(project, "example_image_1.jpg",
[60, 60, 100, 100, 80, 100],
"Personal vehicle")
sa.add_annotation_polyline_to_image(project, "example_image_1.jpg",
[200, 200, 300, 200, 350, 300],
"Personal vehicle")
sa.add_annotation_point_to_image(project, "example_image_1.jpg",
[400, 400], "Personal vehicle")
sa.add_annotation_ellipse_to_image(project, "example_image_1.jpg",
[600, 600, 50, 100, 20],
"Personal vehicle")
sa.add_annotation_template_to_image(
project, "example_image_1.jpg",
[600, 300, 600, 350, 550, 250, 650, 250, 550, 400, 650, 400],
[1, 2, 3, 1, 4, 1, 5, 2, 6, 2], "Human")
sa.add_annotation_cuboid_to_image(project, "example_image_1.jpg",
[60, 300, 200, 350, 120, 325, 250, 500],
"Human")
export = sa.prepare_export(project, include_fuse=True)
(tmpdir / "export").mkdir()
sa.download_export(project, export, (tmpdir / "export"))
# sa.create_fuse_image(
# "./tests/sample_project_vector/example_image_1.jpg",
# "./tests/sample_project_vector/classes/classes.json", "Vector"
# )
paths = sa.download_image(project,
"example_image_1.jpg",
tmpdir,
include_annotations=True,
include_fuse=True,
include_overlay=True)
im1 = Image.open(tmpdir / "export" / "example_image_1.jpg___fuse.png")
im1_array = np.array(im1)
im2 = Image.open(paths[2][0])
im2_array = np.array(im2)
assert im1_array.shape == im2_array.shape
assert im1_array.dtype == im2_array.dtype