def generate_points(img, usesourcemethod=True):
"""
Generate annotation points for the Image img,
and delete any points that had previously existed.
Does nothing if the image already has human annotations,
because we don't want to delete any human work.
"""
# If there are any human annotations for this image,
# abort point generation.
human_annotations = Annotation.objects.filter(image = img).exclude(user = get_robot_user()).exclude(user = get_alleviate_user())
if human_annotations:
return
# Find the annotation area, expressed in pixels.
d = AnnotationAreaUtils.db_format_to_numbers(img.metadata.annotation_area)
annoarea_type = d.pop('type')
if annoarea_type == AnnotationAreaUtils.TYPE_PERCENTAGES:
annoarea_dict = AnnotationAreaUtils.percentages_to_pixels(width=img.original_width, height=img.original_height, **d)
elif annoarea_type == AnnotationAreaUtils.TYPE_PIXELS:
annoarea_dict = d
else:
raise ValueError("Can't generate points with annotation area type '{0}'.".format(annoarea_type))
# Calculate points.
if usesourcemethod:
point_gen_method = img.source.default_point_generation_method
else:
point_gen_method = img.point_generation_method
new_points = calculate_points(
img, annotation_area=annoarea_dict,
**PointGen.db_to_args_format(point_gen_method)
)
# Delete old points for this image, if any.
old_points = Point.objects.filter(image=img)
for old_point in old_points:
old_point.delete()
# Save the newly calculated points.
for new_point in new_points:
Point(row=new_point['row'],
column=new_point['column'],
point_number=new_point['point_number'],
image=img,
).save()
# Update image status.
# Make sure the image goes through the feature-making step again.
status = img.status
status.hasRandomPoints = True
status.save()
def pointgen_check(self, image_id):
"""
Check that an image had annotation points generated as
specified in the point generation method field.
:param image_id: The id of the image to check.
"""
img = Image.objects.get(pk=image_id)
img_width = img.original_width
img_height = img.original_height
pointgen_args = PointGen.db_to_args_format(img.point_generation_method)
points = Point.objects.filter(image=img)
self.assertEqual(points.count(), pointgen_args['simple_number_of_points'])
# Find the expected annotation area, expressed in pixels.
d = AnnotationAreaUtils.db_format_to_numbers(img.metadata.annotation_area)
annoarea_type = d.pop('type')
if annoarea_type == AnnotationAreaUtils.TYPE_PERCENTAGES:
area = AnnotationAreaUtils.percentages_to_pixels(width=img_width, height=img_height, **d)
elif annoarea_type == AnnotationAreaUtils.TYPE_PIXELS:
area = d
elif annoarea_type == AnnotationAreaUtils.TYPE_IMPORTED:
area = dict(min_x=1, max_x=img_width, min_y=1, max_y=img_height)
else:
raise ValueError("Unknown annotation area type.")
if settings.UNIT_TEST_VERBOSITY >= 1:
print "{pointgen_method}".format(
pointgen_method=img.point_gen_method_display(),
)
print "{annotation_area}".format(
annotation_area=img.annotation_area_display(),
)
print "Image dimensions: {width} x {height} pixels".format(
width=img_width, height=img_height,
)
print "X bounds: ({min_x}, {max_x}) Y bounds: ({min_y}, {max_y})".format(
**area
)
for pt in points:
self.assertTrue(area['min_x'] <= pt.column)
self.assertTrue(pt.column <= area['max_x'])
self.assertTrue(area['min_y'] <= pt.row)
self.assertTrue(pt.row <= area['max_y'])
if settings.UNIT_TEST_VERBOSITY >= 1:
print "({col}, {row})".format(col=pt.column, row=pt.row)
def __init__(self, *args, **kwargs):
image = kwargs.pop('image')
if image.metadata.annotation_area:
d = AnnotationAreaUtils.db_format_to_numbers(image.metadata.annotation_area)
annoarea_type = d.pop('type')
if annoarea_type == AnnotationAreaUtils.TYPE_PERCENTAGES:
kwargs['initial'] = AnnotationAreaUtils.percentages_to_pixels(width=image.original_width, height=image.original_height, **d)
elif annoarea_type == AnnotationAreaUtils.TYPE_PIXELS:
kwargs['initial'] = d
elif annoarea_type == AnnotationAreaUtils.TYPE_IMPORTED:
raise ValueError("Points were imported; annotation area should be un-editable.")
super(AnnotationAreaPixelsForm, self).__init__(*args, **kwargs)
self.fields['min_x'] = IntegerField(
label="Left boundary X", required=False,
min_value=1, max_value=image.original_width,
widget=TextInput(attrs={'size': 3})
)
self.fields['max_x'] = IntegerField(
label="Right boundary X", required=False,
min_value=1, max_value=image.original_width,
widget=TextInput(attrs={'size': 3})
)
self.fields['min_y'] = IntegerField(
label="Top boundary Y", required=False,
min_value=1, max_value=image.original_height,
widget=TextInput(attrs={'size': 3})
)
self.fields['max_y'] = IntegerField(
label="Bottom boundary Y", required=False,
min_value=1, max_value=image.original_height,
widget=TextInput(attrs={'size': 3})
)
self.form_help_text = Metadata._meta.get_field('annotation_area').help_text
