Skip to content
This repository has been archived by the owner on Oct 19, 2024. It is now read-only.

Add color profile to training #22

Merged
merged 11 commits into from
Jan 3, 2024
2 changes: 2 additions & 0 deletions ahcore/callbacks/h5_callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,8 @@ def _batch_end(
tile_size=tile_size,
tile_overlap=tile_overlap,
num_samples=num_samples,
color_profile=None,
is_compressed_image=False,
progress=None,
)
new_process = Process(target=new_writer.consume, args=(self.generator(new_queue), child_conn))
Expand Down
Loading