search

bit-bots / YOEO

YouOnlyEncodeOnce - A CNN for Embedded Object Detection and Semantic Segmentation
GNU General Public License v3.0
22 stars 4 forks source link

fix/rescale_segmentation_missing #33

Closed phinik closed 2 years ago

phinik commented 2 years ago 
- implemented "rescale_segmentations(...)" in utils.py (remove padding, interpolate to original image size)
- added rescale_segmentations() call to detect_image() in detect.py

Proposed changes

In detect_image, padding was not removed from the segmentations and they were not rescaled to the original image size

Necessary checks