Here are 2 public repositories matching this topic... YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, n...
SOPHGO-mq also supports QAT quantization for Yolov5. cd ./application/yolov5_example export PYTHONPATH=../../:$PYTHONPATH CUDA_VISIBLE_DEVICES=0 python train.py \ --cfg=yolov5s.yaml \ --weights=yolov5s.pt \ --data=coco.yaml \ --epochs=5 \ --output_path=./ \ --batch_size=8...