Surface small defects are often missed and incorrectly detected due to their small quantity and unapparent visual features.A method named CSYOLOv3,which is based on CutMix and YOLOv3,is proposed to solve such a proble...Surface small defects are often missed and incorrectly detected due to their small quantity and unapparent visual features.A method named CSYOLOv3,which is based on CutMix and YOLOv3,is proposed to solve such a problem.First,a four-image CutMix method is used to increase the small-defect quantity,and the process is dynamically adjusted based on the beta distribution.Then,the classic YOLOv3 is improved to detect small defects accurately.The shallow and large feature maps are split,and several of them are merged with the feature maps of the predicted branch to preserve the shallow features.The loss function of YOLOv3 is optimized and weighted to improve the attention to small defects.Finally,this method is used to detect 512×512 pixel images under RTX 2060Ti GPU,which can reach the speed of 14.09 frame/s,and the mAP is 71.80%,which is 5%-10%higher than that of other methods.For small defects below 64×64 pixels,the mAP of the method reaches 64.15%,which is 14%higher than that of YOLOv3-GIoU.The surface defects of the workpiece can be effectively detected by the proposed method,and the performance in detecting small defects is significantly improved.展开更多
In order to improve the detection accuracy of small objects,a neighborhood fusion-based hierarchical parallel feature pyramid network(NFPN)is proposed.Unlike the layer-by-layer structure adopted in the feature pyramid...In order to improve the detection accuracy of small objects,a neighborhood fusion-based hierarchical parallel feature pyramid network(NFPN)is proposed.Unlike the layer-by-layer structure adopted in the feature pyramid network(FPN)and deconvolutional single shot detector(DSSD),where the bottom layer of the feature pyramid network relies on the top layer,NFPN builds the feature pyramid network with no connections between the upper and lower layers.That is,it only fuses shallow features on similar scales.NFPN is highly portable and can be embedded in many models to further boost performance.Extensive experiments on PASCAL VOC 2007,2012,and COCO datasets demonstrate that the NFPN-based SSD without intricate tricks can exceed the DSSD model in terms of detection accuracy and inference speed,especially for small objects,e.g.,4%to 5%higher mAP(mean average precision)than SSD,and 2%to 3%higher mAP than DSSD.On VOC 2007 test set,the NFPN-based SSD with 300×300 input reaches 79.4%mAP at 34.6 frame/s,and the mAP can raise to 82.9%after using the multi-scale testing strategy.展开更多
基金The National Natural Science Foundation of China(No.52075095).
文摘Surface small defects are often missed and incorrectly detected due to their small quantity and unapparent visual features.A method named CSYOLOv3,which is based on CutMix and YOLOv3,is proposed to solve such a problem.First,a four-image CutMix method is used to increase the small-defect quantity,and the process is dynamically adjusted based on the beta distribution.Then,the classic YOLOv3 is improved to detect small defects accurately.The shallow and large feature maps are split,and several of them are merged with the feature maps of the predicted branch to preserve the shallow features.The loss function of YOLOv3 is optimized and weighted to improve the attention to small defects.Finally,this method is used to detect 512×512 pixel images under RTX 2060Ti GPU,which can reach the speed of 14.09 frame/s,and the mAP is 71.80%,which is 5%-10%higher than that of other methods.For small defects below 64×64 pixels,the mAP of the method reaches 64.15%,which is 14%higher than that of YOLOv3-GIoU.The surface defects of the workpiece can be effectively detected by the proposed method,and the performance in detecting small defects is significantly improved.
基金The National Natural Science Foundation of China(No.61603091)。
文摘In order to improve the detection accuracy of small objects,a neighborhood fusion-based hierarchical parallel feature pyramid network(NFPN)is proposed.Unlike the layer-by-layer structure adopted in the feature pyramid network(FPN)and deconvolutional single shot detector(DSSD),where the bottom layer of the feature pyramid network relies on the top layer,NFPN builds the feature pyramid network with no connections between the upper and lower layers.That is,it only fuses shallow features on similar scales.NFPN is highly portable and can be embedded in many models to further boost performance.Extensive experiments on PASCAL VOC 2007,2012,and COCO datasets demonstrate that the NFPN-based SSD without intricate tricks can exceed the DSSD model in terms of detection accuracy and inference speed,especially for small objects,e.g.,4%to 5%higher mAP(mean average precision)than SSD,and 2%to 3%higher mAP than DSSD.On VOC 2007 test set,the NFPN-based SSD with 300×300 input reaches 79.4%mAP at 34.6 frame/s,and the mAP can raise to 82.9%after using the multi-scale testing strategy.