Method for optimizing depth estimation model, computer device, and storage medium

A computer device comprising:

• a storage device;
• at least one processor;
• and
• the storage device storing one or more programs, which when executed by the at least one processor, cause the at least one processor to:
• obtain a video of an object and capturing a first image and a second image from the video;
• obtain an initial depth estimation model, and obtain an updated depth estimation model by perforating an optimization process on the initial depth estimation model;
• and
• repeatedly perform the optimization process on the updated depth estimation model until the updated depth estimation model meets predetermined requirements, and determine the updated depth estimation model that meets the predetermined requirements as a target depth estimation model;
• wherein the optimization process on the initial depth estimation model comprises:
• obtaining a first depth image of the first image by using the initial depth estimation model;
• obtaining a correspondence between each pixel point in the first image and each pixel point in the second image;
• obtaining a third image by performing a back projection on the first depth image;
• obtaining a fourth image by updating a pixel value of each pixel point in the third image to be a pixel value of the corresponding pixel point in the second image according to the correspondence between each pixel point in the first image and each pixel point in the second image;
• obtaining an initial loss function between the first image and the fourth image based on the initial depth estimation model;
• obtaining a depth value of each pixel point in the first image and the fourth image by using a radar device, and obtaining an updated loss function by optimizing the initial loss function based on the depth value;
• and
• obtaining the updated depth estimation model by optimizing the initial depth estimation model using the updated loss function.

A non-transitory storage medium having stored thereon at least one computer-readable instructions, which when executed by a processor of a computer device, causes the processor to perform a method for optimizing a depth estimation model, wherein the method comprises:

• obtaining a video of an object and capturing a first image and a second image from the video;
• obtaining an initial depth estimation model, and obtaining an updated depth estimation model by performing an optimization process on the initial depth estimation model;
• and
• repeatedly performing the optimization process on the updated depth estimation model until the updated depth estimation model meets predetermined requirements, and determining the updated depth estimation model that meets the predetermined requirements as a target depth estimation model;
• wherein optimization process on the initial depth estimation model comprises:
• obtaining a first depth image of the first image by using the initial depth estimation model;
• obtaining a correspondence between each pixel point in the first image and each pixel point in the second image;
• obtaining a third image by performing back projection on the first depth image;
• obtaining a fourth image by updating a pixel value of each pixel point in the third image to be a pixel value of the corresponding pixel point in the second image according to the correspondence between each pixel point in the first image and each pixel point in the second image;
• obtaining an initial loss function between the first image and the fourth image based on the initial depth estimation model;
• obtaining a depth value of each pixel point in the first image and the fourth image by using a radar device, and obtaining an updated loss function by optimizing the initial loss function based on the depth value;
• and
• obtaining the updated depth estimation model by optimizing the initial depth estimation model using the updated loss function.