Journal of Terahertz Science and Electronic Information Technology , Volume. 20, Issue 8, 810(2022)

Detection of prohibited objects in passive terahertz images based on Focal-EIOU loss function

ZHOU Min*
Author Affiliations
  • [in Chinese]
  • show less

    Aiming at the problem that the image quality fluctuation of passive terahertz security system is caused by environmental change, which affects the recognition algorithm and leads to a significant decrease in accuracy, this paper proposes an improved YOLOv4 algorithm based on Focal-Efficient Intersection Over Union(EIOU) loss function, and uses passive terahertz human security image to conduct model training for prohibited items of knife and gun. A terahertz image database of people carrying suspected objects in different environments and different positions is established, and a rich data set is constructed by image augmentation method. The Complete IOU(CIOU) loss of YOLOv4 is improved to Focal-EIOU loss to improve the robustness of the algorithm for terahertz image recognition, and then a better model is obtained after training. In the test set of this paper, since YOLOv4 algorithm has low robustness for terahertz image recognition accuracy, CIOU loss of YOLOv4 is modified and adjusted to Focal-EIOU loss, and a better model is finally obtained through training. The mean Average Precision (mAP) of the model trained by the improved algorithm reaches 96.4%, the detection speed is about 28 ms, and the average value of IOU is 0.95, which are higher than those of the conventional algorithms under the same conditions, the detection and recognition effect are improved. The experimental results show that the proposed method can effectively improve the suspect identification accuracy of passive terahertz human security system, which is conducive to the popularization and application of this technology in the field of human security.

    Tools

    Get Citation

    Copy Citation Text

    ZHOU Min. Detection of prohibited objects in passive terahertz images based on Focal-EIOU loss function[J]. Journal of Terahertz Science and Electronic Information Technology , 2022, 20(8): 810

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jul. 2, 2021

    Accepted: --

    Published Online: Aug. 30, 2022

    The Author Email: Min ZHOU (zhoumin@crdc.com)

    DOI:10.11805/tkyda2021269

    Topics