利索能及
我要发布
收藏
专利号: 2020107445444
申请人: 平安科技(深圳)有限公司
专利类型:发明专利
专利状态:已下证
更新日期:2026-06-16
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种图像增广方法,其特征在于,包括:

接收模型训练服务器发送的图像增广请求,所述图像增广请求包含图像增广任务集合,所述图像增广任务集合包含至少一个图像增广任务;

判断所述图像增广任务集合中是否同时包含GPU增广任务以及非GPU增广任务;

若所述图像增广任务集合中同时包含GPU增广任务以及非GPU增广任务,从预设的样本库中获取样本图像,并将所述样本图像分发给预设的CPU,以由所述CPU对所述样本图像执行所述非GPU增广任务;

将所述CPU发送的中间处理数据储存到预设的中间数据集合中,所述中间处理数据是所述CPU对所述样本图像执行所述非GPU增广任务后得到的;

从所述中间数据集合中获取中间处理数据,并将中间处理数据分发给预设的GPU,以由所述GPU对所述中间处理数据执行所述GPU增广任务;

将所述GPU发送的第一增广图像数据储存到预设的增广数据集合中,并将所述增广数据集合发送给模型训练服务器,所述第一增广图像数据是所述GPU对所述中间处理数据执行所述GPU增广任务后得到的;

若所述图像增广任务集合中不是同时包含GPU增广任务以及非GPU增广任务,判断所述图像增广任务集合中是否只包含非GPU增广任务;

若所述图像增广任务集合中只包含非GPU增广任务,从预设的样本库中获取样本图像,并将所述样本图像分发给预设的CPU,以由所述CPU对所述样本图像执行所述非GPU增广任务,其中,所述图像增广请求包括所述样本库的存储位置,基于所述存储位置获取所述样本库;

将所述CPU发送的第二增广图像数据储存到预设的增广数据集合中,并将所述增广数据集合发送给模型训练服务器,所述第二增广图像数据是所述CPU对所述样本图像执行所述非GPU增广任务后得到的;

若所述图像增广任务集合中只包含GPU增广任务,从预设的样本库中获取样本图像,并将所述样本图像分发给预设的GPU,以由所述GPU对所述样本图像执行所述GPU增广任务;

将所述GPU发送的第三增广图像数据储存到预设的增广数据集合中,并将所述增广数据集合发送给模型训练服务器,所述第三增广图像数据是所述GPU对所述样本图像执行所述GPU增广任务后得到的。

2.根据权利要求1所述的图像增广方法,其特征在于,所述将所述增广数据集合发送给模型训练服务器,包括:判断模型训练服务器是否部署在本地;

若模型训练服务器部署在本地,通过共享内存的方式将所述增广数据集合发送给模型训练服务器。

3.根据权利要求2所述的图像增广方法,其特征在于,所述将所述增广数据集合发送给模型训练服务器,还包括:若模型训练服务器不是部署在本地,通过RPC传输或者HTTP传输的方式将所述增广数据集合发送给模型训练服务器。

4.根据权利要求1所述的图像增广方法,其特征在于,所述CPU包括多个CPU执行线程,所述将所述样本图像分发给预设的CPU,包括:获取所述CPU中处于空闲状态的CPU执行线程作为目标CPU执行线程;

向所述目标CPU执行线程分配样本图像。

5.根据权利要求1所述的图像增广方法,其特征在于,所述GPU包括多个GPU执行线程,所述将中间处理数据分发给预设的GPU,包括:获取所述GPU中处于空闲状态的GPU执行线程作为目标GPU执行线程;

向所述目标GPU执行线程分配中间处理数据。

6.一种图像增广装置,其特征在于,包括:

接收单元,用于接收模型训练服务器发送的图像增广请求,所述图像增广请求包含图像增广任务集合,所述图像增广任务集合包含至少一个图像增广任务;

第一判断单元,用于判断所述图像增广任务集合中是否同时包含GPU增广任务以及非GPU增广任务;

第一分发单元,用于若所述图像增广任务集合中同时包含GPU增广任务以及非GPU增广任务,从预设的样本库中获取样本图像,并将所述样本图像分发给预设的CPU,以由所述CPU对所述样本图像执行所述非GPU增广任务;

第一存储单元,用于将所述CPU发送的中间处理数据储存到预设的中间数据集合中,所述中间处理数据是所述CPU对所述样本图像执行所述非GPU增广任务后得到的;

第二分发单元,用于从所述数据集合中获取中间处理数据,并将中间处理数据分发给预设的GPU,以由所述GPU对所述中间处理数据执行所述GPU增广任务;

第一发送单元,用于将所述GPU发送的第一增广图像数据储存到预设的增广数据集合中,并将所述增广数据集合发送给模型训练服务器,所述第一增广图像数据是所述GPU对所述中间处理数据执行所述GPU增广任务后得到的;

第三判断单元,用于若所述图像增广任务集合中不是同时包含GPU增广任务以及非GPU增广任务,判断所述图像增广任务集合中是否只包含非GPU增广任务;

第三分发单元,用于若所述图像增广任务集合中只包含非GPU增广任务,从预设的样本库中获取样本图像,并将所述样本图像分发给预设的CPU,以由所述CPU对所述样本图像执行所述非GPU增广任务,其中,所述图像增广请求包括所述样本库的存储位置,基于所述存储位置获取所述样本库;

第四发送单元,用于将所述CPU发送的第二增广图像数据储存到预设的增广数据集合中,并将所述增广数据集合发送给模型训练服务器,所述第二增广图像数据是所述CPU对所述样本图像执行所述非GPU增广任务后得到的;

第四分发单元,用于若所述图像增广任务集合中只包含GPU增广任务,从预设的样本库中获取样本图像,并将所述样本图像分发给预设的GPU,以由所述GPU对所述样本图像执行所述GPU增广任务;

第五发送单元,用于将所述GPU发送的第三增广图像数据储存到预设的增广数据集合中,并将所述增广数据集合发送给模型训练服务器,所述第三增广图像数据是所述GPU对所述样本图像执行所述GPU增广任务后得到的。

7.一种计算机设备,其特征在于,所述计算机设备包括存储器及处理器,所述存储器上存储有计算机程序,所述处理器执行所述计算机程序时实现如权利要求1‑5中任一项所述的方法。

8.一种计算机可读存储介质,其特征在于,所述存储介质存储有计算机程序,所述计算机程序当被处理器执行时可实现如权利要求1‑5中任一项所述的方法。