Skip to content

Commit

Permalink
another indent and update comments
Browse files Browse the repository at this point in the history
  • Loading branch information
HaleySchuhl committed May 14, 2024
1 parent d8d2f54 commit 57bb762
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions plantcv/annotate/classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,15 +244,14 @@ def correct_mask(self, bin_img, bin_img_recover, coords):
(x, y) = self.coords[names][id]
unrec_points.append((x, y))
new_name = str(names) + "_unrecovered"
# Pull out unrecovered coords into new class
# Put unrecovered coords into new class
self.coords[new_name] = unrec_points
# Overwrite class with only coords that have corresponding objects in the completed_mask
# Overwrite attribute, only coords that have corresponding objects in the completed_mask
new_points = []
for i, (x, y) in enumerate(self.coords[names]):
if i not in unrecovered_ids:
new_points.append((x, y))

self.coords[names] = new_points
self.coords[names] = new_points

completed_mask1 = 1*((completed_mask + 1*(completed_mask == 255)) != 0).astype(np.uint8)

Expand Down

0 comments on commit 57bb762

Please sign in to comment.