zacario-li / Segmentation-based-deep-learning-approach-for-surface-defect-detection

Segmentation-based deep-learning approach for surface-defect detection with pytorch
MIT License
22 stars 2 forks source link

[TODO]Even iteration take defective image, odd iteration take non-defective image sampling strategy #3

Open zacario-li opened 4 years ago

zacario-li commented 4 years ago

In paper, author use "Even iteration take defective image, odd iteration take non-defective image sampling strategy", I will try to implement it later.

DCC-lzhy commented 3 years ago

这个好解决的,可以仿照TensorFlow版本的,搞两个dataloader,针对Positive和Negtive,调用iter()后,用next()获得batch。 。

zacario-li commented 3 years ago

这个好解决的,可以仿照TensorFlow版本的,搞两个dataloader,针对Positive和Negtive,调用iter()后,用next()获得batch。 。

欢迎你提交个PR过来,我会merge进去