jizhishutong/YOLOU

似乎没有关闭马赛克增强

jie311 opened this issue · 1 comments

if mode == 'yolox':
if epoch == epochs - no_aug_epochs:
LOGGER.info("--->No mosaic aug now!")
dataset.mosaic = False # close mosaic
LOGGER.info("--->Add additional L1 loss now!")

我自己单步调试的时候发现, dataset.mosaic 在运行过程中还是True