david8862 / keras-YOLOv3-model-set

end-to-end YOLOv4/v3/v2 object detection pipeline, implemented on tf.keras with different technologies
MIT License
640 stars 222 forks source link

Mosaic #169

Open guishilike opened 3 years ago

guishilike commented 3 years ago

mosaic 的实现方式和yolov5略有不同,应当是拼接四个图片(可能有缺失部分),而不是针对相对位置进行裁剪

david8862 commented 3 years ago

mosaic 的实现方式和yolov5略有不同,应当是拼接四个图片(可能有缺失部分),而不是针对相对位置进行裁剪

@guishilike 目前的Mosaic augmentation实现是按照YOLOv4论文中描述的方式。后续会考虑增加YOLOv5版本Mosaic augmentation的接口

guishilike commented 3 years ago

按照repo的数据增强方式,会产生大量填充区域,yolov5在做Mosaic之前并未填充。

david8862 commented 3 years ago

按照repo的数据增强方式,会产生大量填充区域,yolov5在做Mosaic之前并未填充。

目前新增的mosaic_v5 augment暂时是按照本项目原有的数据处理流程,即先进行单帧图片的letterbox resize及各种基本增强之后再进行拼接,故letterbox填充的边界会影响到mosaic的拼接效果。后续如果有机会的话会考虑进行修正