search

junqiangchen / LiTS---Liver-Tumor-Segmentation-Challenge

LiTS - Liver Tumor Segmentation Challenge
MIT License
234 stars 73 forks source link

Performance issues in LiTS/data_input/dataset_input.py(P2) #28

Closed DLPerf closed 2 years ago

DLPerf commented 3 years ago

Hello,I found a performance issue in the definition of datagenerator , LiTS/data_input/dataset_input.py, dataset = dataset.map(_parse_function) was called without num_parallel_calls. I think it will increase the efficiency of your program if you add this.

The same issues also exist in dataset = dataset.map(_preprocess_function)

Here is the documemtation of tensorflow to support this thing.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

junqiangchen commented 3 years ago

thank you,if you can fix the problem that is good.