search

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

bb_targets and mask_targets may be unbound #121

Open Ertz012 opened 7 months ago

Ertz012 commented 7 months ago

https://github.com/bit-bots/YOEO/blob/0aaecc2359c619db8f8666b0f3890044308578ba/yoeo/utils/datasets.py#L143C9-L143C55

jaagut commented 7 months ago

On the way I thought about it, and I think making transforms optional is just not necessary. I believe in almost all cases one wants to apply some transforms, at least the default ones. If transforms is not necessary, we leave out the if check and the variables cannot be unbound anymore

Flova commented 7 months ago

I am not sure what the exact problem is? Could you provide additional information? Is it related to batching/transforms when a batch consists only of negative data?