diff --git a/README.md b/README.md index 22ce7f80..db4acdb0 100644 --- a/README.md +++ b/README.md @@ -237,3 +237,8 @@ python tools/infer.py --weights yolov6s6.pt --img 1280 --source img.jpg / imgdir ### [FAQ(Continuously updated)](https://github.com/meituan/YOLOv6/wiki/FAQ%EF%BC%88Continuously-updated%EF%BC%89) + +If you have any questions, welcome to join our WeChat group to discuss and exchange. +

+ +

\ No newline at end of file diff --git a/README_cn.md b/README_cn.md index b2294c5f..c6740730 100644 --- a/README_cn.md +++ b/README_cn.md @@ -213,3 +213,8 @@ python tools/infer.py --weights yolov6s6.pt --img 1280 --source img.jpg / imgdir ### [FAQ(持续更新)](https://github.com/meituan/YOLOv6/wiki/FAQ%EF%BC%88Continuously-updated%EF%BC%89) + +如果您有任何问题,欢迎加入我们的微信群一起讨论交流! +

+ +

\ No newline at end of file diff --git a/yolov6/core/engine.py b/yolov6/core/engine.py index 5a37b477..8dfda3e8 100644 --- a/yolov6/core/engine.py +++ b/yolov6/core/engine.py @@ -55,8 +55,8 @@ def __init__(self, args, cfg, device): self.distill_ns = True if self.args.distill and self.cfg.model.type in ['YOLOv6n','YOLOv6s'] else False model = self.get_model(args, cfg, self.num_classes, device) if self.args.distill: - if self.args.fuse_ab and self.distill_ns: - LOGGER.error('ERROR in: Distill n/s models should turn off the fuse_ab.\n') + if self.args.fuse_ab: + LOGGER.error('ERROR in: Distill models should turn off the fuse_ab.\n') exit() self.teacher_model = self.get_teacher_model(args, cfg, self.num_classes, device) if self.args.quant: