PSPNet-TF-Reproduce

PSPNet-TF-Reproduce copied to clipboard

Binary segmentation on a custom dataset

7

Hi, I am new to segmentation and trying to use this repo to train the network from scratch on a custom dataset having only 2 classes (binary segmentation) and so...

cspearl

Performance issue in /database (by P3)

1

Hello! I've found a performance issue in /reader.py: `dataset.batch(batch_size)`[(here)](https://github.com/holyseven/PSPNet-TF-Reproduce/blob/6923c09bd6ef8b1195a90253bc5b2329f78231db/database/reader.py#L310) should be called before `.map(_training_data_preprocess, num_parallel_calls=batch_size)`[(here)](https://github.com/holyseven/PSPNet-TF-Reproduce/blob/6923c09bd6ef8b1195a90253bc5b2329f78231db/database/reader.py#L308), which could make your program more efficient. Here is [the tensorflow document](https://tensorflow.google.cn/guide/data_performance?hl=zh_cn#vectorized_mapping) to support it....

DLPerf