add Negative Data training and testing

bit-bots / YOEO

YouOnlyEncodeOnce - A CNN for Embedded Object Detection and Semantic Segmentation

GNU General Public License v3.0

21 stars 4 forks source link

Open Ertz012 opened 7 months ago

Ertz012 commented 7 months ago

added a feature that trains and tests on negative data this should reduce false positive segmentations on background

Ertz012 commented 7 months ago

TODO: split negative data in training and test

jaagut commented 7 months ago

Todo: make superclass dataset with params (e g. Size) and collate_fn. Thus, it's defined on every dataset, not just ListDataset