yolov5_post 组件应该使用MODULE_PROCESS_REGISTRY 注册，而非MODULE_ZOO_REGISTRY

ModelTC / United-Perception

United Perception

Apache License 2.0

428 stars 65 forks source link

Open wangshankun opened 7 months ago

wangshankun commented 7 months ago

yolov5_post 组件应该使用MODULE_PROCESS_REGISTRY 注册，而非MODULE_ZOO_REGISTRY；

如果使用MODULE_ZOO_REGISTRY，那么YOLOv5在执行构建量化模型的时候会将后处理一并当成backbone处理，但fx机制明显不支持后处理网络的量化，这样会让YOLOv5无法量化