angelolab / Nimbus

Other
12 stars 1 forks source link

Added resize argument and functionality to SegmentationTFRecords #38

Closed JLrumberger closed 1 year ago

JLrumberger commented 1 year ago

What is the purpose of this PR?

Prepare half resolution tfrecord datasets and use SegmentationTFRecords on Erins dataset where image suffixes are ".tif" and not ".tiff".

How did you implement your changes

Added to new arguments to SegmentationTFRecords: resize and img_suffix. Resizing is done with cv2.resize in SegmentationTFRecords.prepare_example and SegmentationTFRecords.get_inst_binary_masks. Suffixes are just added wherever images are loaded from disk.

Remaining issues None