Hello! I've found a performance issue in /model.py: .batch(batch_size)(here) should be called before .map(im_read, num_parallel_calls=num_parallel_calls)(here) and .map(get_patches_fn, num_parallel_calls=num_parallel_calls)(here), which could make your program more efficient.
Here is the tensorflow document to support it.
Besides, you need to check the function get_patches_fn called in .map(get_patches_fn, num_parallel_calls=num_parallel_calls) whether to be affected or not to make the changed code work properly. For example, if get_patches_fn needs data with shape (x, y, z) as its input before fix, it would require data with shape (batch_size, x, y, z) after fix.
Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.
Hello! I've found a performance issue in /model.py:
.batch(batch_size)(here) should be called before.map(im_read, num_parallel_calls=num_parallel_calls)(here) and.map(get_patches_fn, num_parallel_calls=num_parallel_calls)(here), which could make your program more efficient.Here is the tensorflow document to support it.
Besides, you need to check the function
get_patches_fncalled in.map(get_patches_fn, num_parallel_calls=num_parallel_calls)whether to be affected or not to make the changed code work properly. For example, ifget_patches_fnneeds data with shape (x, y, z) as its input before fix, it would require data with shape (batch_size, x, y, z) after fix.Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.