The invention relates to the field of digital image processing, and relates to an accident prediction model training method comprising the following steps: obtaining a training data set which comprises a plurality of image frames collected by an automobile data recorder; constructing a network structure of the accident prediction model; the training data set is used to train a network of the accident prediction model, in each training, a loss function of the accident prediction model to be trained is determined according to the features of the image frames and an accident prediction result output by the accident prediction model to be trained, and the loss function comprises consistency comparison loss and prediction loss; and optimizing network parameters of the to-be-trained accident prediction model according to the loss function to obtain a final accident prediction model. The invention further provides an accident prediction model training device, equipment and a storage medium. The method can improve the understanding capability of the traffic accident model for the image frame, and improves the accuracy of the accident prediction result.

    本发明涉及数字图像处理的领域,涉及一种事故预测模型训练方法,包括:获取训练数据集,所述数据训练集包括行车记录仪采集的若干图像帧;构建事故预测模型的网络结构;利用训练数据集对事故预测模型的网络进行训练,在每次训练中,根据图像帧的特征和待训练事故预测模型输出的事故预测结果确定待训练事故预测模型的损失函数,所述损失函数包括一致性对比损失和预测损失;根据损失函数优化待训练事故预测模型的网络参数,得到最终的事故预测模型。本发明还提出一种事故预测模型训练装置、设备以及存储介质。本发明能够提高交通事故模型对图像帧的理解能力,提高事故预测结果的准确性。


    Access

    Download


    Export, share and cite



    Title :

    Accident prediction model training method and device, equipment and storage medium


    Additional title:

    一种事故预测模型训练方法、装置、设备及存储介质


    Contributors:
    ZHU DONG (author) / FANG XIANGMING (author) / SONG WEN (author) / TANG GUOMEI (author) / ZHANG JIAN (author) / HU XIAODONG (author) / YAN GE (author) / ZHONG YUANHONG (author)

    Publication date :

    2025-01-14


    Type of media :

    Patent


    Type of material :

    Electronic Resource


    Language :

    Chinese


    Classification :

    IPC:    G06V / G06N COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS , Rechnersysteme, basierend auf spezifischen Rechenmodellen / B60W CONJOINT CONTROL OF VEHICLE SUB-UNITS OF DIFFERENT TYPE OR DIFFERENT FUNCTION , Gemeinsame Steuerung oder Regelung von Fahrzeug-Unteraggregaten verschiedenen Typs oder verschiedener Funktion / G08G Anlagen zur Steuerung, Regelung oder Überwachung des Verkehrs , TRAFFIC CONTROL SYSTEMS



    Traffic accident prediction method and device, equipment and storage medium

    HOU CHEN | European Patent Office | 2020

    Free access

    Traffic accident prediction method and device, electronic equipment and storage medium

    LIU WEI / ZHANG TAO / LU YISHENG et al. | European Patent Office | 2023

    Free access

    Training method of traffic accident prediction model and traffic accident prediction method and device

    WANG NIANMING / CHEN YANG / ZHOU MINGKE et al. | European Patent Office | 2024

    Free access


    Vehicle track prediction model training method and device, equipment and storage medium

    CUI MAOYUAN / WANG CHAO / HAN JIAQI et al. | European Patent Office | 2023

    Free access