EP4595001A1 - Neural segmentation fields for representing three-dimensional scenes - Google Patents
Neural segmentation fields for representing three-dimensional scenesInfo
- Publication number
- EP4595001A1 EP4595001A1 EP23777460.9A EP23777460A EP4595001A1 EP 4595001 A1 EP4595001 A1 EP 4595001A1 EP 23777460 A EP23777460 A EP 23777460A EP 4595001 A1 EP4595001 A1 EP 4595001A1
- Authority
- EP
- European Patent Office
- Prior art keywords
- scene
- neural network
- image
- processor
- segmentation map
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
Classifications
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T15/00—Three-dimensional [3D] image rendering
- G06T15/10—Geometric effects
- G06T15/20—Perspective computation
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/09—Supervised learning
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T15/00—Three-dimensional [3D] image rendering
- G06T15/08—Volume rendering
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/11—Region-based segmentation
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
Definitions
- NEURAL SEGMENTATION FIELDS FOR REPRESENTING THREE-DIMENSIONAL SCENES 1.
- This application claims the benefit of priority from U.S. Provisional Application Serial No.63/410,344 filed on 27 September 2022, which is incorporated by reference herein in its entirety.
- Field of the Disclosure [0002] Various example embodiments relate to rendering a three-dimensional (3D) scene into a two-dimensional (2D) space.
- 3D Three-dimensional
- 2D two-dimensional
- semantic segmentation every pixel of the image is typically associated with a corresponding class label. Semantic segmentation differs from object detection because, unlike object detection, semantic segmentation works at the pixel level to determine relatively accurate contours of objects within an image. [0004] Unlike semantic segmentation, which treats multiple objects within a class as one entity, instance segmentation identifies individual objects within classes. Instance segmentation is sometimes considered to be a refined version of semantic segmentation. Various computer vision applications use either semantic segmentation or a combination of semantic and instance segmentation. BRIEF SUMMARY OF SOME SPECIFIC EMBODIMENTS [0005] Disclosed herein are various embodiments of methods and apparatus for rendering segmentation maps of a 3D scene using machine learning.
- a smaller neural network is trained to render segmentation maps corresponding to arbitrarily 1 selected views of the 3D scene, with the training being performed using a larger neural network pretrained to represent the 3D scene in color texture and volume density.
- the small neural network is configured to work on top of the pretrained larger neural network, thereby providing the capability to obtain segmentation information faster and with relatively low additional load on the computational resources.
- an entropy-based loss function and regularization terms therefor which are constructed to facilitate the training process for the small neural network, e.g., by taking advantage of the auxiliary information available in conjunction with or from the previous training of the larger neural network.
- an image-processing method comprising training, via a processor, a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene, the training including: computing, with the processor, color texture and volume density corresponding to a selected training view of the 3D scene, the computing being performed using a 3D representation pretrained to represent the 3D scene; generating, with the processor, a predicted segmentation map corresponding to the selected training view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and adjusting, with the processor, configuration parameters of NN nodes of the first neural network based on a loss function configured to receive, as a first input thereof, a ground-truth segmentation map corresponding to the selected training view and further configured to receive, as a second input thereof, the predicted segmentation map.
- a non-transitory computer- readable medium storing instructions that, when executed by the processor, cause the processor to perform operations comprising the above or below image-processing method.
- an image-processing apparatus comprising: at least one processor; and at least one memory including program code; and wherein the at least one memory and the program code are configured to, with the at least one processor, cause the apparatus at least to train a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene; and wherein to train the first neural network, the apparatus is configured to: compute color texture and volume density corresponding to a selected training view of the 3D scene, the computing being performed using a 3D representation pretrained to represent the 3D scene; generate a predicted segmentation map corresponding to the 2 selected training view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and adjust configuration parameters of NN nodes of the first neural
- an image-processing method comprising testing, via a processor, a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene, the testing including: computing, with the processor, color texture and volume density corresponding to a selected view of the 3D scene, the computing being performed using a 3D representation pretrained to represent the 3D scene; and generating, with the processor, a segmentation map corresponding to the selected view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and wherein the first neural network has been trained, with the processor, using the pretrained 3D representation.
- an image-processing apparatus comprising: at least one processor; and at least one memory including program code; wherein the at least one memory and the program code are configured to, with the at least one processor, cause the apparatus at least to test a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene; wherein to test the first neural network, the apparatus is configured to: compute color texture and volume density corresponding to a selected view of the 3D scene, the computing being performed using a 3D representation pretrained to represent the 3D scene; and generate a segmentation map corresponding to the selected view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and wherein the first neural network has been trained using the pretrained 3D representation.
- FIGs.1A-1B are block diagrams illustrating example process flows for training and testing, respectively, of a neural segmentation field according to various embodiments.
- FIG.2 is a block diagram illustrating a multilayer perceptron (MLP) that can be used to implement a neural segmentation field according to an embodiment.
- FIG.3 is a flowchart illustrating a process of training the MLP of FIG.2 according to various embodiments.
- FIG.4 is a flowchart illustrating a process of testing the MLP of FIG.2 according to various embodiments.
- FIG.5 is a flowchart illustrating computations of a loss function used in the process of FIG.3 according to some examples.
- FIG.6 is a flowchart illustrating computations of an example regularization term for the loss function used in the process of FIG.3 according to some examples.
- FIG.7 is a flowchart illustrating computations of another example regularization term for the loss function used in the process of FIG.3 according to some examples.
- FIG.8 is a block diagram illustrating a computing device according to an embodiment.
- a neural network is a typically nonlinear trainable circuit comprising a plurality of processing elements (PEs), also referred to as “neurons,” “artificial neurons,” or “NN nodes.”
- PEs processing elements
- a neural network is an application-specific circuit, wherein different PEs are implemented as respective configurable sub-circuits connected by physical links forming the corresponding physical network.
- a neural network may be computer emulated, in which cases one or more electronic processors are programmed to perform signal processing analogous to that of the corresponding application-specific circuit.
- Each PE of a neural network typically has connections with one or more other PEs. The plurality of connections between the (physical or computer-emulated) PEs defines the neural network’s topology.
- the PEs may be aggregated into layers. Different layers may have different types of PEs configured to perform different respective kinds of transformations on their inputs. Signals may travel from the first PE layer (typically referred to as the input layer) to the last PE layer (typically referred to as the output layer).
- the neural network may have one or more intermediate PE layers (typically referred to as the hidden layers) located between the input and output PE layers.
- An example PE may scale, sum, and bias the incoming signals and use an activation function to produce an output signal that is a static nonlinear function of the biased sum. The resulting PE output may become either one of the neural network’s outputs or be sent to one or more other PEs through the corresponding connection(s).
- a Neural Radiance Field represents a 3D scene implicitly, e.g., using a neural network that takes a 3D position and viewing direction as inputs and generates the corresponding predicted color texture and volume density as outputs.
- the corresponding neural network can be trained using a set of 2D images with known camera poses and pertinent intrinsic information.
- a NeRF is configured to implicitly represent a scene using the continuous 3D scene density ⁇ and color ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ as a function of continuous five-dimensional (5D) input vectors of spatial coordinates ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ and viewing directions ⁇ ⁇ ⁇ ⁇ , ⁇ , where ⁇ , ⁇ , ⁇ are the Cartesian coordinates, and ⁇ , ⁇ are the angles used in the spherical coordinate system.
- 5D continuous five-dimensional
- the continuous function that converts a 5D input ⁇ ⁇ , ⁇ to a four-dimensional (4D) output ⁇ ⁇ , ⁇ , ⁇ , ⁇ is approximated using an neural network, such as a multilayer perceptron (MLP).
- MLP multilayer perceptron
- the density ⁇ is a function of only ⁇
- the radiance ⁇ is a function of both position ⁇ and viewing direction ⁇ .
- the color of a pixel in a 2D rendering of the scene is computed via volume rendering.
- the neural network Given a set of images of a 3D scene with known camera parameters, the neural network’s PE parameters can be optimized via gradient descent by minimizing photometric differences between the rendered and ground-truth images.
- An MLP is a fully connected feedforward neural network that has an input layer, one or more hidden layers, and an output layer.
- the “fully connected” attribute means that there is a respective weighted connection between each PE from the previous layer to each PE of the adjacent subsequent layer.
- An entry of the corresponding weight matrix ⁇ is ⁇ ⁇ , which includes weights for the connections between the a-th PE of the subsequent layer and the b-th PE of the previous layer.
- the output ⁇ ⁇ ⁇ ⁇ thereof can be expressed as a function ⁇ of the weighted sum of input values ⁇ ⁇ ⁇ ⁇ , where ⁇ ⁇ is the number of PEs in the 6 subsequent layer, and ⁇ ⁇ is the number of PEs in the previous layer.
- the function ⁇ is the above- mentioned activation function.
- a nonlimiting example of the activation function f is the ReLU (Rectified Linear Unit) function expressed as follows: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ for ⁇ ⁇ 0 0 for ⁇ ⁇ 0 (2)
- Other suitable activation functions can be used as well.
- a NeRF may be configured to additionally map the inputs to a higher dimensional space using a series of trigonometric functions ⁇ and then pass the higher dimensional inputs to the neural network for better fitting output data with high-frequency components.
- the series ⁇ is defined as follows: ⁇ ⁇ ⁇ ⁇ sin ⁇ 2 ⁇ ⁇ ⁇ , cos ⁇ 2 ⁇ ⁇ ⁇ , ... , sin ⁇ 2 ⁇ ⁇ ⁇ , cos ⁇ 2 ⁇ ⁇ ⁇ (3)
- L is the number of to each coordinate in ⁇ ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ and to each component in the viewing direction ⁇ ⁇ ⁇ ⁇ , ⁇ .
- the coordinates and viewing directions are typically normalized to lie in the interval [ ⁇ 1, 1]. Note that Eq. (3) describes an example implementation of positional encoding.
- NeRF represents a 3D scene as the volume density and directional emitted radiance at any point of the 3D space and renders the color of any ray passing through the scene using volume rendering by numerical integration.
- the approximated expected color can be computed using the following equation: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ (4) where ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ exp ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ; ⁇ ⁇ ⁇ 1 ⁇ exp ⁇ ⁇ ; and ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ is the distance ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ to represent [0031] In addition to the color texture, some applications and/or users require and/or expect semantic segmentation of the 3D scene, which the above-described NeRF does not provide.
- various embodiments disclosed herein address at least this problem in the state of the art by providing a neural segmentation field implemented as an additional smaller neural network trained and configured to implicitly represent semantic segmentation of the same 3D scene using the initial pretrained neural network, e.g., the NeRF.
- the training of the smaller neural network takes advantage of the fact that the semantics, color texture, and geometry of a 3D scene are typically correlated to a significant degree.
- the neural segmentation field can be learned in a relatively fast and efficient manner and then the corresponding smaller neural network can be used to render 2D segmentation maps of the 3D scene corresponding to arbitrarily selected viewpoints.
- the training process implement enhancements, e.g., in the form of one or more regularization terms, that can improve the segmentation quality for at least some 3D scenes.
- an example embodiment of the disclosed neural segmentation field beneficially achieves a relatively high segmentation quality (e.g., mIoU > 0.9) with a relatively small model size (e.g., ⁇ 0.4 MB) for a relatively large number (e.g., > 25) of semantic classes.
- Various embodiments of the neural segmentation field can be applied, e.g., to object editing (such as color manipulation or enhancement) and/or to 3D-scene encoding and compression.
- object editing such as color manipulation or enhancement
- 3D-scene encoding and compression e.g., the acronym “mIoU” stands for mean intersection-over-union.
- FIGs.1A-1B are block diagrams illustrating example process flows (100, 101) for training and testing, respectively, of a neural segmentation field (110) according to various embodiments.
- a training stage (108) of the training process flow (100) produces the neural segmentation field (110), which is an implicit segmentation representation that represents segmentation in a 3D scene as a neural network, based on a 3D-scene representation (102), a set of 2D segmentation maps (104), and corresponding camera parameters (106).
- the 3D-scene representation (102) is a NeRF.
- the 3D-scene representation (102) is another suitable 3D scene representation pretrained to represent the 3D scene.
- such a 3D scene representation can be a non-NN-based solution, e.g., Spherical Harmonics or another suitable implementation thereof.
- semantic segmentation and instance segmentation can be represented by the neural segmentation field (110).
- a testing stage (114) of the testing process flow (101) produces 2D segmentation maps (116) corresponding to testing (e.g., new or arbitrary) viewpoints using the 3D-scene representation (102), the neural segmentation field (110) produced in the training process flow (100), and camera parameters (112) for the testing viewpoints.
- Specific nonlimiting examples of the training stage (108) and the testing stage (114) are described in more detail below, e.g., in reference to FIGs.3 and 4.
- FIG.2 is a block diagram illustrating an MLP (200) that can be used to implement a neural segmentation field according to an embodiment.
- the MPL (200) has four layers (2101-2104).
- the first layer (210 1 ) is the input layer.
- the next two layers (210 2 , 210 3 ) are the hidden layers.
- the fourth layer (210 4 ) is the output layer.
- MLP (200) can have N hidden layers, where N is a positive integer.
- N is in the range from 1 to 4.
- M PEs (202) where M is a positive integer.
- M is in the range from 16 to 256.
- Each of the PEs (20221-2022M) of the second layer (2102) is directly connected to receive respective inputs from each of the PEs (202 11 -202 1M ) of the input layer (210 1 ).
- Each of the PEs (202 31 -202 3M ) of the third layer (210 3 ) is directly connected to receive respective inputs from each of the PEs (20221-2022M) of the second layer (2102).
- Each of the PEs (20241-2024M) of the output layer (2104) is directly connected to receive respective inputs from each of the PEs (202 31 -202 3M ) of the third layer (210 3 ).
- Similar PE/layer connections are implemented for additional embodiments corresponding to other values of 9 N.
- each of the PEs (202) operates using the ReLU activation function (see Eq. (2)). In other examples, other suitable activation functions can also be used.
- the inputs to the MLP (200) include the positional encoded 3D coordinates ⁇ ⁇ , ⁇ ⁇ , ⁇ ⁇ (also see Eq. (3)) as well as a 4D input including the radiance ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ and the density ⁇ .
- the output of the MLP (200) is a plurality of semantic logits ⁇ ⁇ ⁇ ⁇ , where ⁇ ⁇ is the number of segmentation classes.
- FIG.3 shows a flowchart (300) illustrating a process of training the MLP (NN 1, 200) according to various embodiments.
- Such training is performed using camera parameters (302) of a set of training views.
- Each of the training views is associated (342) with a corresponding ground- truth label map (340), which serves as one of the inputs to a loss function (330).
- Iterative adjustments (328) of the PE parameters used in the MLP (200) are performed to optimize (e.g., approximately minimize) the loss function (330) over the set of training views. Once optimization criteria are met, the training process is terminated, and the corresponding “optimal” PE parameters are fixed and subsequently used in the testing (working) mode of operation, which is described in more detail below in reference to FIG.4.
- the camera parameters (302) are used to obtain a 5D input (304, 306) for a pretrained neural network (NN 2, 310).
- the neural network (310) implements a NeRF, e.g., as described above.
- the 5D input (304, 306) includes the corresponding 3D spatial coordinates ⁇ ⁇ , ⁇ , ⁇ (306) and the 2D viewing direction ⁇ ⁇ , ⁇ (304).
- the pretrained neural network (310) In response to the 5D input (304, 306), the pretrained neural network (310) generates a 4D output ⁇ ⁇ , ⁇ , ⁇ , ⁇ (314).
- Positional encoding (312) is applied to the 3D spatial coordinates ⁇ ⁇ , ⁇ , ⁇ (306) to generate a corresponding series ⁇ ⁇ (318) (e.g., see Eq. (3)).
- the 4D output ⁇ ⁇ , ⁇ , ⁇ , ⁇ (314) of the neural network (310) together with the series ⁇ ⁇ (318) provide an input for the MLP (200), which is being trained.
- the MLP (200) generates a plurality of semantic logits ⁇ ⁇ ⁇ ⁇ (320).
- Volume rendering (322) which can be implemented, e.g., in a manner consistent with Eq.
- FIG.4 shows a flowchart (400) illustrating a process of testing the MLP (NN 1, 200) according to various embodiments.
- the term “testing” refers to the working mode of operation using which semantic segmentation maps corresponding to arbitrary views of the 3D scene can be generated.
- the MLP (200) Prior to entering the working mode of operation, the MLP (200) is trained as described above in reference to FIG.3.
- the PE parameters of the MLP (200) are constant, e.g., fixed at the optimal values determined using the training process (300).
- the pretrained neural network (310) remains in the same configuration in the testing process (400) as in the training process (300).
- the corresponding camera parameters (402) are used to obtain the corresponding 5D input (304, 306) for the neural network (310).
- the neural network (310) In response to the 5D input (304, 306), the neural network (310) generates the 4D output ⁇ ⁇ , ⁇ , ⁇ , ⁇ (314).
- Positional encoding (312) is applied to the 3D spatial coordinates ⁇ ⁇ , ⁇ , ⁇ (306) to generate the corresponding series ⁇ ⁇ ⁇ ⁇ (318).
- the 4D output ⁇ ⁇ , ⁇ , ⁇ , ⁇ (314) of the neural network (310) together with the series ⁇ ⁇ ⁇ ⁇ (318) provide an input for the trained MLP (200).
- the trained MLP (200) generates the plurality of semantic logits ⁇ ⁇ ⁇ ⁇ (320).
- Subsequent processing of the MLP’s output includes applying a softmax function (426) to the semantic logits (320) to obtain the probabilities ⁇ of rays/pixels belonging to the class ⁇ .
- the probability of being in the class ⁇ can be computed as follows: ⁇ ⁇ ⁇ , ⁇ , ⁇ ⁇ , ⁇ , ⁇ ⁇ (6)
- S h,w,c is the probabilities for the pixels representing the new view make up a probability map (428).
- the probability map (428) is further processed to generate a label map (430), wherein different objects within the new view are delineated and labeled (e.g., using color coding) in accordance with the respective classes c thereof.
- the loss function (330) is defined as the cross entropy between the predicted semantic segmentation map (324) and the ground-truth segmentation map (340), e.g., as follows: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ , ⁇ ⁇ , ⁇ , ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ ⁇ ⁇ , ⁇ ⁇ ⁇ log ⁇ , ⁇ ⁇ ⁇ ⁇ ⁇ log ⁇ ⁇ , ⁇ , ⁇ (7) ⁇ ⁇ ⁇ , ⁇ , ⁇ ( 8) where which is the target label; ⁇ ⁇ is the total number of classes c, and ⁇ , ⁇ , ⁇ is the ground-truth label for ⁇ , ⁇ , ⁇ ⁇ ⁇ 0, 1 ⁇ at pixel (h, w) of the map (340) for the class ⁇ .
- the loss function (330) is defined as the weighted cross entropy between the predicted semantic segmentation map (324) and the ground-truth segmentation map (340), e.g., as follows: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ , ⁇ ⁇ ⁇ ⁇ ⁇ l ⁇ , ⁇ ⁇ , ⁇ , ⁇ ⁇ , ⁇ ⁇ ⁇ , ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ og ⁇ , ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ log ⁇ ⁇ ⁇ , ⁇ , ⁇ (9) where ⁇ be defined as follows: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ exp ⁇ ⁇ ⁇ (11) where ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ 0, 1 ⁇ is the magnitude of an image edge in the rendered 2D RGB image (332).
- FIG.5 shows a flowchart (500) illustrating computations of the weighted cross entropy for the loss function (330) according to some examples.
- Inputs to the computations (500) include the predicted semantic segmentation map (324), the ground-truth segmentation map (340), and the 12 RGB image (332).
- Edge detection (504) is performed for the RGB image (332) to detect object edges therein.
- the weighting function w of Eq. (11) is then computed (502) such that the object- edge pixels are given higher weights than object-body pixels.
- a deep-learning- based edge detector can be used to implement the computation blocks (502, 504).
- the loss function (330) is then computed in accordance with Eqs. (9)-(10). [0045]
- the loss function (330) includes one or more weighted regularization terms added to the cross entropy of Eq.
- Eq. (12) provides an example of the loss function (330) employing such regularization terms.
- pixelwise regularization and pairwise regularization are example regularization terms; ⁇ ⁇ and ⁇ ⁇ are constants whose values can be empirically selected.
- ⁇ ⁇ and ⁇ ⁇ can be 0.01 and 0.001, respectively.
- One example regularization term implements pixelwise edge regularization.
- the corresponding unary regularization term ⁇ ⁇ ⁇ ⁇ at pixel ⁇ is defined as: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ (13) where ⁇ ⁇ is the segmentation edge map; and ⁇ ⁇ is the edge detected, e.g., using the above- described edge detector.
- the map ⁇ ⁇ is computed as gradient magnitude over the segmentation probability map, which is expressed as follows: ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ where ⁇ ⁇ ⁇ . ⁇ and ⁇ ⁇ ⁇ . ⁇ probability map of the c-th class.
- FIG.6 shows a flowchart (600) illustrating computations of a pixelwise regularization term (610) for the loss function (330) according to some examples.
- Inputs to the computations (600) include the predicted semantic segmentation map (324) and the RGB image (332).
- Edge detection (504) is performed for the RGB image (332) to detect object edges therein.
- An edge map (604) is generated based on the object edges detected in the block (504).
- a segmentation edge map (602) is generated based on the predicted semantic segmentation map (324).
- the pixelwise edge regularization term (610) is then computed in accordance with Eq. (13) based on the edge maps (602, 604).
- Another example regularization term implements pairwise edge regularization.
- the corresponding pairwise regularization term ⁇ ⁇ , ⁇ for pixels ⁇ and q is defined as: ⁇ ⁇ , ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ , ⁇ , ⁇ ⁇ ⁇ ⁇ , ⁇ (16) where ⁇ ⁇ , ⁇ ⁇ are the (332) and on the predicted segmentation map (324), respectively; ⁇ is a function configured to penalize large segmentation distances for the pixel pairs whose RGB distance is small.
- ⁇ ⁇ can be provided by the pretrained neural network (310).
- the pairwise edge regularization term is configured to use ⁇ ⁇ for improving ⁇ ⁇ and thus ⁇ .
- ⁇ ⁇ is the length of the shortest path ⁇ between two pixels ⁇ and ⁇ on a weighted graph ⁇ ⁇ , which is a 4-neighbor graph with pixels as nodes and the RGB image-edge magnitude as the edge weight.
- the corresponding mathematical expression is as follows: ⁇ ⁇ ⁇ ⁇ , ⁇ ⁇ ⁇ ⁇ , ⁇ ⁇ ⁇ ⁇ ⁇ (18) where ⁇ ⁇ , ⁇ is the shortest path between ⁇ and ⁇ on ⁇ ⁇ computed using the Dijkstra’s algorithm; and ⁇ ⁇ ⁇ ⁇ , ⁇ is the length of ⁇ ⁇ , ⁇ . If there are image edges between two pixels on ⁇ ⁇ , then ⁇ ⁇ is large.
- FIG.7 shows a flowchart (700) illustrating computations of a pairwise edge regularization term (710) for the loss function (330) according to some examples.
- Inputs to the computations (700) include the predicted semantic segmentation map (324) and the RGB image (332).
- Edge detection (504) is performed for the RGB image (332) to detect object edges therein.
- An edge map (604) is generated based on the object edges detected in the block (504).
- a pairwise distance map (702) is generated in accordance with Eq. (18) based on the edge map (604).
- a pairwise distance map (704) is generated based the predicted semantic segmentation map (324) and using the above definition of ⁇ ⁇ ⁇ ⁇ , ⁇ .
- the pairwise edge regularization term (710) is then computed in accordance with Eq. (16) based on the distance maps (702, 704).
- the pairwise edge regularization term (710) may be calculated over a local region surrounding the pixel in question, the local area having the size of, e.g., 8 ⁇ 8 or 16 ⁇ 16 pixels.
- FIG.8 is a block diagram illustrating a computing device (800) according to an embodiment.
- the device (800) can be used, e.g., to implement the process flows (100, 101).
- the device (800) comprises input/output (I/O) devices (810), an image-processing engine (IPE, 820), and a memory (830).
- the I/O devices (810) may be used to enable the device (800) to receive at least the parameters of the 3D representation (102) and to output at least the parameters of the neural segmentation field (110) and the segmentation maps (116).
- the I/O devices (810) may also be used to connect the device (800) to a display.
- the memory (830) may have buffers to receive image data corresponding to a 3D scene to be rendered.
- the image data may be, e.g., in the form of one or more image files.
- the memory (830) may provide parts of the data to the IPE (820) for processing therein.
- the IPE (820) includes a processor (822) and a memory (824).
- the memory (824) may store therein program code, which when executed by the processor (822) enables the IPE (820) to perform image processing, including but not limited to the image processing in accordance with some or all of the above-described flowcharts (100, 101, 300, 400, 500, 600, 700).
- the program code may include, inter alia, the program code used to emulate the various neural networks, e.g., the NN 1 (200) and the NN 2 (310) described above.
- the IPE (820) may perform rendering processing thereof and provide the corresponding viewable image(s) for 15 being viewed on the display.
- the viewable image can be, e.g., in the form of a suitable image file outputted through the I/O devices (810).
- an image- processing method comprising training, via a processor, a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene, the training including: computing, with the processor, color texture and volume density corresponding to a selected training view of the 3D scene, the computing being performed using a second neural network pretrained to represent the 3D scene; generating, with the processor, a predicted segmentation map corresponding to the selected training view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and adjusting, with the processor, configuration parameters of processing elements of the first neural network based on a loss function configured to receive, as a first input thereof, a ground-truth segmentation map corresponding to the selected training view and further configured to receive, as a second input thereof, the predicted segmentation map.
- the first neural network is a multilayer perceptron; and wherein the second neural network is another multilayer perceptron that implements a neural radiance field.
- the first neural network is configured to output semantic logits corresponding to the selected training view of the 3D scene in response to said color texture and said volume density.
- training further includes: calculating a cross entropy between the predicted segmentation map and the ground-truth segmentation map; and obtaining a loss-function value using the cross entropy.
- the loss function is configured to receive, as a third input thereof, a color image of the 3D scene corresponding to the selected training view. 16 [0058] In some embodiments of any of the above methods, the training further includes: detecting object edges in the color image; assigning different weights to object-edge pixels and to object-body pixels; calculating a cross entropy between the predicted segmentation map and the ground-truth segmentation map using the different weights; and obtaining a loss-function value using the cross entropy. [0059] In some embodiments of any of the above methods, the loss function includes one or more regularization terms favoring alignment of corresponding edges in the predicted segmentation map and the ground-truth segmentation map.
- the generating is performed in response to positional encoding of a 3D spatial coordinate corresponding to the selected training view.
- the image-processing method further comprises creating, with the processor, a segmentation map corresponding to an arbitrarily selected view of the 3D scene, the creating being performed using the first neural network after the training.
- the image-processing method further comprises: applying, with the processor, a softmax function to the segmentation map corresponding to the arbitrarily selected view of the 3D scene to generate a probability map; and converting, with the processor, the probability map into a semantic label map representing the arbitrarily selected view of the 3D scene.
- a non- transitory computer-readable medium storing instructions that, when executed by the processor, cause the processor to perform operations comprising any of the above methods.
- an image-processing apparatus comprising: at least one processor; and at least one memory including program code; and wherein the at least one memory and the program code are configured to, with the at least one processor, cause the apparatus at least to train a first neural network to render 17 segmentation maps corresponding to arbitrarily selected views of a 3D scene; and wherein to train the first neural network, the apparatus is configured to: compute color texture and volume density corresponding to a selected training view of the 3D scene, the computing being performed using a second neural network pretrained to represent the 3D scene; generate a predicted segmentation map corresponding to the selected training view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and adjust configuration parameters of processing elements of the first neural network based on a loss function receiving, as a first input thereof, a ground-
- the first neural network is a multilayer perceptron; and wherein the second neural network is another multilayer perceptron that implements a neural radiance field.
- the first neural network is configured to output semantic logits corresponding to the selected training view of the 3D scene in response to said color texture and said volume density.
- the apparatus is further configured to: calculate a cross entropy between the predicted segmentation map and the ground-truth segmentation map; and obtain a loss-function value using the cross entropy.
- the loss function is configured to receive, as a third input thereof, a color image of the 3D scene corresponding to the selected training view.
- the apparatus is further configured to: detect object edges in the color image; assign different weights to object-edge pixels and to object-body pixels; calculate a cross entropy between the predicted segmentation map and the ground-truth segmentation map using the different weights; and obtain a loss-function value using the cross entropy.
- the loss function includes one or more regularization terms favoring alignment of corresponding edges in the predicted segmentation map and the ground-truth segmentation map.
- the at least one memory and the program code are further configured to, with the at least one processor, cause the apparatus to create a segmentation map corresponding to an arbitrarily selected view of the 3D scene, the creating being performed using the first neural network after the training.
- the at least one memory and the program code are further configured to, with the at least one processor, cause the apparatus to: apply a softmax function to the segmentation map corresponding to the arbitrarily selected view of the 3D scene to generate a probability map; and convert the probability map into a semantic label map representing the arbitrarily selected view of the 3D scene.
- an image-processing method comprising testing, via a processor, a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene, the testing including: computing, with the processor, color texture and volume density corresponding to a selected view of the 3D scene, the computing being performed using a 3D representation pretrained to represent the 3D scene; and generating, with the processor, a segmentation map corresponding to the selected view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and wherein the first neural network has been trained, with the processor, using the pretrained 3D representation.
- an image-processing apparatus comprising: at least one processor; and at least one memory including program code; wherein the at least one memory and the program code are configured to, with the at least one processor, cause the apparatus at least to test a first neural network to render segmentation maps corresponding to arbitrarily selected views of a 3D scene; wherein to test the first neural network, the apparatus is configured to: compute color texture and volume density corresponding to 19 a selected view of the 3D scene, the computing being performed using a 3D representation pretrained to represent the 3D scene; and generate a segmentation map corresponding to the selected view of the 3D scene, the generating being performed using the first neural network based on said color texture and said volume density; and wherein the first neural network has been trained using the pretrained 3D representation.
- Some embodiments can also be embodied in the form of program code recorded in tangible media, such as magnetic recording media, optical recording media, solid state memory, floppy diskettes, CD-ROMs, hard drives, or any other non-transitory machine-readable storage medium, wherein, when the program code is loaded into and executed by a machine, such as a computer, the machine becomes an apparatus for practicing various embodiments described herein.
- Some embodiments can also be embodied in the form of program code, for example, stored in a non-transitory machine-readable storage medium including being loaded into and/or executed by a machine, wherein, when the program code is loaded into and executed by a machine, such as a computer or a processor, the machine becomes an apparatus for practicing various embodiments described herein.
- the phrase “if it is determined” or “if [a stated condition] is detected” may be construed to mean “upon determining” or “in response to determining” or “upon detecting [the stated condition or event]” or “in response to detecting [the stated condition or event].”
- the terms “couple,” “coupling,” “coupled,” “connect,” “connecting,” or “connected” refer to any manner known in the art or later developed in which energy is allowed to be transferred between two or more elements, and the interposition of 22 one or more additional elements is contemplated, although not required. Conversely, the terms “directly coupled,” “directly connected,” etc., imply the absence of such additional elements.
- processors may be provided through the use of dedicated hardware as well as hardware capable of executing software in association with appropriate software.
- the functions may be provided by a single dedicated processor, by a single shared processor, or by a plurality of individual processors, some of which may be shared.
- processor or “controller” should not be construed to refer exclusively to hardware capable of executing software, and may implicitly include, without limitation, digital signal processor (DSP) hardware, network processor, application specific integrated circuit (ASIC), field programmable gate array (FPGA), read only memory (ROM) for storing software, random access memory (RAM), and nonvolatile storage. Other hardware, conventional and/or custom, may also be included.
- DSP digital signal processor
- ASIC application specific integrated circuit
- FPGA field programmable gate array
- ROM read only memory
- RAM random access memory
- nonvolatile storage nonvolatile storage.
- Other hardware conventional and/or custom, may also be included.
- any switches shown in the figures are conceptual only. Their function may be carried out through the operation of program logic, through dedicated logic, through the interaction of program control and dedicated logic, or even manually, the particular technique being selectable by the implementer as more specifically understood from the context.
- circuit may refer to one or more or all of the following: (a) hardware-only circuit implementations (such as implementations in only analog and/or digital circuitry); (b) combinations of hardware circuits and software, such as (as applicable): (i) a combination of analog and/or digital hardware circuit(s) with software/firmware and (ii) any portions of hardware processor(s) with software (including digital signal processor(s)), software, and memory(ies) that work together to cause an apparatus, such as a mobile phone or server, to perform various functions); and (c) hardware circuit(s) and or processor(s), such as a microprocessor(s) or a portion of a microprocessor(s), that requires software (e.g., firmware) for operation, but the software may not be present when it is not needed for operation.”
- This definition of circuitry applies to all uses of this term in this application, including in any claims.
- circuitry also covers an implementation of merely a hardware circuit or processor (or multiple processors) or portion of a hardware circuit or processor and its (or their) accompanying software and/or firmware.
- circuitry also covers, for 23 example and if applicable to the particular claim element, a baseband integrated circuit or processor integrated circuit for a mobile device or a similar integrated circuit in server, a cellular network device, or other computing or network device.
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Computing Systems (AREA)
- General Health & Medical Sciences (AREA)
- Life Sciences & Earth Sciences (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Biomedical Technology (AREA)
- Molecular Biology (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Biophysics (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Computer Graphics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Geometry (AREA)
- Image Analysis (AREA)
Abstract
Description
Claims
Applications Claiming Priority (2)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| US202263410344P | 2022-09-27 | 2022-09-27 | |
| PCT/US2023/073251 WO2024073214A1 (en) | 2022-09-27 | 2023-08-31 | Neural segmentation fields for representing three-dimensional scenes |
Publications (1)
| Publication Number | Publication Date |
|---|---|
| EP4595001A1 true EP4595001A1 (en) | 2025-08-06 |
Family
ID=88204120
Family Applications (1)
| Application Number | Title | Priority Date | Filing Date |
|---|---|---|---|
| EP23777460.9A Pending EP4595001A1 (en) | 2022-09-27 | 2023-08-31 | Neural segmentation fields for representing three-dimensional scenes |
Country Status (4)
| Country | Link |
|---|---|
| EP (1) | EP4595001A1 (en) |
| JP (1) | JP7853519B2 (en) |
| CN (1) | CN120266156A (en) |
| WO (1) | WO2024073214A1 (en) |
Families Citing this family (1)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| US12307589B2 (en) * | 2023-03-21 | 2025-05-20 | Qualcomm Incorporated | Generating semantically-labelled three-dimensional models |
Family Cites Families (3)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| EP4115392A4 (en) | 2020-03-04 | 2023-08-23 | Magic Leap, Inc. | Systems and methods for efficient floorplan generation from 3d scans of indoor scenes |
| US12518472B2 (en) | 2020-11-16 | 2026-01-06 | Google Llc | Deformable neural radiance fields |
| EP4298607B1 (en) * | 2021-02-24 | 2025-05-07 | Google LLC | Color and infra-red three-dimensional reconstruction using implicit radiance function |
-
2023
- 2023-08-31 CN CN202380081505.2A patent/CN120266156A/en active Pending
- 2023-08-31 EP EP23777460.9A patent/EP4595001A1/en active Pending
- 2023-08-31 JP JP2025518342A patent/JP7853519B2/en active Active
- 2023-08-31 WO PCT/US2023/073251 patent/WO2024073214A1/en not_active Ceased
Also Published As
| Publication number | Publication date |
|---|---|
| JP7853519B2 (en) | 2026-04-28 |
| WO2024073214A1 (en) | 2024-04-04 |
| JP2025531498A (en) | 2025-09-19 |
| CN120266156A (en) | 2025-07-04 |
Similar Documents
| Publication | Publication Date | Title |
|---|---|---|
| US11798173B1 (en) | Moving point detection | |
| CN109584248B (en) | Infrared target instance segmentation method based on feature fusion and dense connection network | |
| EP2731075B1 (en) | Backfilling points in a point cloud | |
| US20220381914A1 (en) | Systems and methods for sparse convolution of unstructured data | |
| US12602898B2 (en) | Neural semantic fields for generalizable semantic segmentation of 3D scenes | |
| CA3235728A1 (en) | Co-learning object and relationship detection with density aware loss | |
| Liu et al. | RGB-D joint modelling with scene geometric information for indoor semantic segmentation | |
| dos Santos Rosa et al. | Sparse-to-continuous: Enhancing monocular depth estimation using occupancy maps | |
| CN110781744A (en) | A small-scale pedestrian detection method based on multi-level feature fusion | |
| JP7831486B2 (en) | Signal processing device and signal processing method | |
| CN114219942A (en) | Apparatus and method for training a generative neural network | |
| Dou et al. | Analysis of vehicle and pedestrian detection effects of improved YOLOv8 model in drone-assisted urban traffic monitoring system | |
| WO2024238237A1 (en) | Planar mesh reconstruction using images from multiple camera poses | |
| Schöntag et al. | Towards cross domain transfer learning for underwater correspondence search | |
| EP4595001A1 (en) | Neural segmentation fields for representing three-dimensional scenes | |
| Tighkhorshid et al. | Car depth estimation within a monocular image using a light CNN: A. Tighkhorshid et al. | |
| Sheorey et al. | Uncertainty estimation for KLT tracking | |
| Le et al. | Simple linear iterative clustering based low-cost pseudo-LiDAR for 3D object detection in autonomous driving | |
| US20250182460A1 (en) | Refining image features and/or descriptors | |
| US20250378569A1 (en) | Determining positions of objects based on images and ground-plane models | |
| Li et al. | 2.5 D-VoteNet: Depth Map based 3D Object Detection for Real-Time Applications. | |
| WO2024002534A1 (en) | Method and system of selecting one or more images for human labelling | |
| Liu et al. | MotionRFCN: Motion segmentation using consecutive dense depth maps | |
| CN120047677B (en) | UAV target detection method and device | |
| US20250292553A1 (en) | Determining error for training computer-vision models |
Legal Events
| Date | Code | Title | Description |
|---|---|---|---|
| STAA | Information on the status of an ep patent application or granted ep patent |
Free format text: STATUS: UNKNOWN |
|
| STAA | Information on the status of an ep patent application or granted ep patent |
Free format text: STATUS: THE INTERNATIONAL PUBLICATION HAS BEEN MADE |
|
| PUAI | Public reference made under article 153(3) epc to a published international application that has entered the european phase |
Free format text: ORIGINAL CODE: 0009012 |
|
| STAA | Information on the status of an ep patent application or granted ep patent |
Free format text: STATUS: REQUEST FOR EXAMINATION WAS MADE |
|
| 17P | Request for examination filed |
Effective date: 20250409 |
|
| AK | Designated contracting states |
Kind code of ref document: A1 Designated state(s): AL AT BE BG CH CY CZ DE DK EE ES FI FR GB GR HR HU IE IS IT LI LT LU LV MC ME MK MT NL NO PL PT RO RS SE SI SK SM TR |
|
| P01 | Opt-out of the competence of the unified patent court (upc) registered |
Free format text: CASE NUMBER: UPC_APP_6005_4595001/2025 Effective date: 20250904 |
|
| DAV | Request for validation of the european patent (deleted) | ||
| DAX | Request for extension of the european patent (deleted) |