diff --git a/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py b/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py index 07142048bc9ea9067ce945c8b7055c5e57d85fe4..9433552ed3d600e843272311efda5d855e3f5938 100644 --- a/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py +++ b/extensions/chaeo/examples/transfer_labels_to_ilastik_object_classifier.py @@ -168,4 +168,11 @@ if __name__ == '__main__': result_acc, _ = mod.infer(train_zstack_raw, train_zstack_mask) write_accessor_data_to_file(where_patch_stack / 'result.tif', result_acc) - print(where_patch_stack / 'result.tif') \ No newline at end of file + print(where_patch_stack / 'result.tif') + + # write comparison + train_labels = generate_file_accessor(where_patch_stack / 'zstack_train_label.tif') + write_accessor_data_to_file( + where_patch_stack / 'comp.tif', + InMemoryDataAccessor(result_acc.data == train_labels.data) + ) \ No newline at end of file