WO2024254055A1 - Image segmentation and feature extraction using neural networks and a spatio-temporal filter - Google Patents
Image segmentation and feature extraction using neural networks and a spatio-temporal filter Download PDFInfo
- Publication number
- WO2024254055A1 WO2024254055A1 PCT/US2024/032388 US2024032388W WO2024254055A1 WO 2024254055 A1 WO2024254055 A1 WO 2024254055A1 US 2024032388 W US2024032388 W US 2024032388W WO 2024254055 A1 WO2024254055 A1 WO 2024254055A1
- Authority
- WO
- WIPO (PCT)
- Prior art keywords
- segmentation
- feature
- maps
- map
- spatial resolution
- 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.)
- Ceased
Links
Classifications
-
- 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
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/174—Segmentation; Edge detection involving the use of two or more images
-
- 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/10—Image acquisition modality
- G06T2207/10016—Video; Image sequence
-
- 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/20024—Filtering details
- G06T2207/20028—Bilateral filtering
-
- 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
- the present invention relates generally to images. More particularly, an embodiment of the present invention relates to techniques for image segmentation and feature extraction using a spatiotemporal filter.
- Image segmentation and feature extraction are key image processing operations in a variety of image and video applications, including computer vision, image coding (compression), and display management. Given an input image, such operations will output semantic or geometric information about a scene in the image. Without limitation, the output of such operations may be denoted, interchangeably, as a “feature map” or as a “segmentation map.” Given a feature map or a segmentation map, in some embodiments, one may perform a thresholding operation, thus generating a corresponding binary “feature mask” or “segmentation mask.” In recent years, segmentation techniques based on artificial intelligence and deep learning were also examined.
- the term “deep learning” refers to neural networks having at least three layers, and preferably more than three layers.
- neural networks in order to address computational complexity issues, it is common to operate in low-resolution images and up-sample the output; however, classical up-sampling techniques, such as bilinear or bicubic interpolation, might create additional artifacts, such as blurring, thus compromising the quality of the output.
- classical up-sampling techniques such as bilinear or bicubic interpolation
- image-segmentation and feature extraction techniques are developed.
- FIG.1 depicts a processing pipeline for image segmentation and feature extraction according to an example embodiment of the present invention
- FIG.2 depicts examples of the Gaussian kernels being used in spatiotemporal filtering according to an example embodiment of the present invention
- FIG.3 depicts examples of feature maps for a video frame.
- Example embodiments described herein relate to methods for image segmentation for video sequences.
- a processor receives a sequence of pictures in a video sequence at a high spatial resolution.
- the processor applies (105) a feature extraction neural network (NN) to the current picture to generate at the high spatial resolution: a first set of one or more feature-maps (107) extracted from early layers of the feature extraction NN, and an output set of one or more feature-maps (109) from the feature extraction NN; generates (110) in a low spatial resolution a spatially down-sampled version of the output set of feature maps; applies (115) a segmentation neural network to the down-sampled version of the output set of feature maps to generate a segmentation map (117) at the low spatial resolution; generates (120) a source segmentation map (122) at the high spatial resolution by applying a spatial up-sampling to the segmentation map; generates filter-parameters for a spatiotemporal filter based on the first set of feature-maps (107); and applies the spatiotemporal filter to the source segmentation map to generate an output segmentation map (127).
- a feature extraction neural network NN
- FIG.1 depicts an example processing pipeline for image segmentation and feature extraction according to an embodiment.
- input to the process is a sequence of video images (102) in high resolution.
- each frame is processed by a feature-extraction neural network (105), followed by a segmentation neural network (115).
- segmentation and feature extraction may be performed using only the current frame and T prior frames.
- the output (109) of feature extraction (105) is typically followed directly by a segmentation network (110), which processes data at the same spatial resolution as the output of the feature network.
- a spatial down-sampling step (110) between the two networks, so segmentation is performed at a lower spatial resolution (say, 1 ⁇ 2, 1 ⁇ 4, or 1/8, and the like, of the input spatial resolution).
- such sub-sampling can be performed either by pooling layers or strided convolutions at the back-end of feature extraction NN (105) or at the front-end of segmentation NN (115), or one can use any known sub-sampling techniques known in the art.
- the feature extraction and segmentation networks (105, 115) can be a series of convolutional neural networks (CNNs) trained either through the whole processing pipeline or they can be pretrained CNNs. For example, in an embodiment, feature extraction was performed using the CNN model VGG19 in Ref. [1-2]. Segmentation can be performed by any known in the art method (e.g., see Ref. [3]).
- the intermediate output (107) of neural network 105 serve as a first good feature extractor to be used later to improve the output of a segmentation NN.
- the first few layers extract more shallow low level features (e.g., edges, texture, color, and the like) and the deeper layers extract more semantic features (e.g., is this a human or a plant).
- the final segmentation map is usually extracted after many layers (the deeper part of the network) since that will enable it to have more information regarding what objects are present in the image.
- Segmentation networks may be trained as a per pixel classification algorithm.
- ground truth labels for a binary algorithm e.g., one with two classes: object and background
- the ground truth labels for a binary algorithm would be a per pixel map with 0s for the background and 1s for the foreground.
- a spatiotemporal filter processes the following inputs: • High resolution feature map data (107), from feature-extraction NN (105) • Source segmentation-map data (122), the up-sampled source-map-output of the segmentation NN (115) The output (127) of this spatiotemporal filter represents the final output segmentation map at the original high resolution.
- up-sampling filter 120 may use any known in the art filter, like bilinear or bicubic interpolation, spline interpolation, and the like, or in an embodiment, it can be part of the segmentation NN 115.
- a conventional bilateral filter can be expressed as ⁇ ⁇ ⁇ 1 ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ 1 ⁇ ⁇ ⁇ ⁇ ⁇
- ⁇ denotes the filtered image
- ⁇ is the coordinate of the current pixel being filtered
- ⁇ and ⁇ are pixel values at pixel locations ⁇ and ⁇ of the image to be filtered
- ⁇ ⁇ and ⁇ are Gaussian functions that measure affinity in space and intensity respectively and are calculated as follows, 1 '( ⁇ % ⁇ ⁇ ⁇ &# ⁇ () ⁇ 2 ⁇ Variables " ⁇ and " ⁇ denote the
- a colorization or style transfer operator can be applied to a low resolution version of the image and the original image can be used as a guide to upsample the low-resolution transformed image.
- a binary segmentation algorithm for an input image, one can get a pixelwise confidence mask with values 0-1 (e.g., 0 denotes low confidence and 1 denotes very high confidence that a certain object is present).
- a novel spatiotemporal filter (125) is proposed that utilizes three distinct components: 1. Spatial and temporal distance 2. Texture and color similarity, drawn from network feature maps of the original image and input images over time 3.
- this joint spatiotemporal trilateral up-sampling filter is described as: ⁇ 12,3 ⁇ ⁇ ⁇ 1 ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ 5 ⁇ ⁇ ⁇ ⁇ ⁇ 5 ⁇ 6 ⁇ ⁇ 89 ⁇ ⁇ ⁇ -.: ⁇ ⁇ ⁇ -.: 9; ⁇ 5 ⁇ ⁇ ⁇ 4 .7 ⁇ where ⁇ ⁇ .7 and ⁇ 4 are Gaussian functions as defined in eq.
- ⁇ and " ⁇ are now the standard deviations in the spatial and temporal axes, which allows us to control the affinity functions along the axes individually.
- ⁇ ⁇ is calculated as, ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ M$NO ⁇ ⁇ 7M$NO ⁇ ⁇ [00024]
- window ⁇ is now of size M x M x (2T+1) , where (2T+1) denotes the number of frames used in the temporal domain including the anchor (current) frame, typically around 3 or 7. This corresponds to a temporal window that includes: T frames before the current frame, the current frame, and T frames after the current frame.
- the spatial window size M can be similar to the size used in the spatial bilateral filter.
- N is the number of feature layers (to be explained more later).
- a 3D Gaussian kernel could be split into separable 2D or even 1D Gaussian functions for increased computational efficiency. Because of the use of a temporal window with 2T+1 frames, to generate the source feature maps (122) more efficiently the processing steps of feature extraction (105), segmentation (115), downsampling (110), and up-sampling (120) can be duplicated 2T+1 times, thus allowing for parallel processing.
- the input to spatio-temporal filtering (125) includes 2T+1 source-maps and (2T+1) x N feature maps, and generates an output feature frame for the current input frame.
- Spatiotemporal distance kernel ( ⁇ ⁇ ⁇ [00025] From equation (6), the spatiotemporal distance kernel, ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ ⁇ , is a 3D kernel whose sigma values (standard deviation in the Gaussian for space and time may be tuned separately.
- a sigma value for space (“ ⁇ ) should be a standard deviation around 5 pixels when the segmentation map is being upsampled four times (4x) (e.g., it should be larger when a higher upsampling scale is used), and a sigma value for time ( " ⁇ ) should be a standard deviation around 1/4 second (converted to frames at 24fps; 5 frames – an odd number -maintains uniformity).
- ⁇ acts as a sort of edge detector in an image. It compares the pixel p along a window and takes the magnitude of this difference. Traditionally this is done on intensity (RGB or Y differences). More advanced methods would include using a perceptually uniform color space. However, these methods would still fail when presented with two objects of similar color. [00028]
- filtering is using the high-resolution, low-level feature maps (107) from the initial layers of a CNN based neural network (105) (hence it is denoted as “sigma-f “ for “feature”) to construct our Gaussian function.
- FIG.3 depicts two examples (305, 310) of feature maps extracted from an input video frame, the two examples picked at random from the 128 output channels at layer 4 of VGG19. [00029] For improved performance and efficiency, the feature maps are extracted from the segmentation network itself.
- the proposed method can be implemented in a more parallel fashion which can improve speed and efficiency. For example, if the system benefits from lots of memory storage, one method to improve speed is to save the feature ⁇ ⁇ "NO ) and source kernels ⁇ ⁇ ⁇ computed from previously processed images.
- the spatiotemporal kernel is this can be saved for all frames to use. Saving previously used kernels allows, on the current frame, to simply compute the kernels for the current frame alone and directly multiply the older kernels together. The speedup would be directly proportional to the time window one is using. [00035] In a system that benefits from parallelization, all frames within a time window can be computed at once. Since only the final multiplication depends on other frames, the majority of the complex processing can be done in parallel. [00036] If a system that is best at sequential processing, one may be able to utilize smaller parallelization efforts. Since the feature maps are coming from the early layers of the segmentation machine learning algorithm, the feature maps are generated before the source- segmentation map is complete.
- Embodiments of the present invention may be implemented with a computer system, systems configured in electronic circuitry and components, an integrated circuit (IC) device such as a microcontroller, a field programmable gate array (FPGA), or another configurable or programmable logic device (PLD), a discrete time or digital signal processor (DSP), an application specific IC (ASIC), and/or apparatus that includes one or more of such systems, devices or components.
- IC integrated circuit
- FPGA field programmable gate array
- PLD configurable or programmable logic device
- DSP discrete time or digital signal processor
- ASIC application specific IC
- the computer and/or IC may perform, control, or execute instructions related to image transformations, such as those described herein.
- the computer and/or IC may compute any of a variety of parameters or values that relate image- segmentation processes described herein.
- the image and video embodiments may be implemented in hardware, software, firmware and various combinations thereof.
- Certain implementations of the invention comprise computer processors which execute software instructions which cause the processors to perform a method of the invention.
- processors in a display, an encoder, a set top box, a transcoder or the like may implement methods related to image-segmentation processes as described above by executing software instructions in a program memory accessible to the processors.
- the invention may also be provided in the form of a program product.
- the program product may comprise any tangible and non-transitory medium which carries a set of computer-readable signals comprising instructions which, when executed by a data processor, cause the data processor to execute a method of the invention.
- Program products according to the invention may be in any of a wide variety of tangible forms.
- the program product may comprise, for example, physical media such as magnetic data storage media including floppy diskettes, hard disk drives, optical data storage media including CD ROMs, DVDs, electronic data storage media including ROMs, flash RAM, or the like.
- the computer-readable signals on the program product may optionally be compressed or encrypted.
- a method for generating segmentation maps of pictures in a video sequence comprising: receiving a sequence of video pictures in a first spatial resolution; for a current picture (102) in the sequence of video pictures: applying (105) a feature extraction neural network (NN) to the current picture to generate at the first spatial resolution: a first set of one or more feature-maps (107) extracted from early layers of the feature extraction NN, and an output set of one or more feature-maps (109) from the feature extraction NN; generating (110) in a second spatial resolution a spatially down-sampled version of the output set of feature maps, wherein the second spatial resolution is lower than the first spatial resolution; applying (115) a segmentation neural network to the down-sampled version of the output set of feature maps to generate a segmentation map (117) at the second spatial resolution; generating (120) a source segmentation map (122) at the first spatial resolution by applying a spatial up-sampling to the segmentation map; generating filter-parameters for a spatiotemporal filter based on the first spatial
- EEE 2. The method of EEE 1, wherein the second spatial resolution is one-half, one-fourth, or one-eighth of the first spatial resolution.
- EEE 3. The method of EEE 1 or EEE 2, wherein the filter parameters for the spatiotemporal filter comprise a spatiotemporal distance kernel ( ⁇ ⁇ ⁇ , a feature map range kernel ⁇ "NO ), and a segmentation confidence kernel ⁇ ⁇ ⁇ 4 ⁇ .
- EEE 5 The method of EEE 3 or EEE 4, wherein ⁇ ⁇ comprises a multivariate Gaussian defined as, ⁇ ⁇ ⁇ ⁇ ⁇ > C %F F HF ( $ %( ' G 'I , where d, defined as, " # ⁇ 0 0 " ⁇ and " ⁇ are standard deviations in the spatial and temporal axes.
- EEE 6. The method of any of EEEs 3-5, wherein ⁇ ⁇ .7 and ⁇ ⁇ 4 are Gaussian functions ⁇ ⁇ ⁇ ⁇ 1 '( %& # ⁇ () 2 !# $ with standard deviations, ", EEE 7.
- EEE 9 An apparatus comprising a processor and configured to perform any one of the methods recited in EEEs 1-7.
- EEE 9. A non-transitory computer-readable storage medium having stored thereon computer-executable instruction for executing a method with one or more processors in accordance with any one of the EEEs 1-7.
Landscapes
- Engineering & Computer Science (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- Image Analysis (AREA)
Abstract
Description
Claims
Priority Applications (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| EP24732816.4A EP4724982A1 (en) | 2023-06-09 | 2024-06-04 | Image segmentation and feature extraction using neural networks and a spatio-temporal filter |
Applications Claiming Priority (4)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| US202363507181P | 2023-06-09 | 2023-06-09 | |
| US63/507,181 | 2023-06-09 | ||
| EP23189251 | 2023-08-02 | ||
| EP23189251.4 | 2023-08-02 |
Publications (1)
| Publication Number | Publication Date |
|---|---|
| WO2024254055A1 true WO2024254055A1 (en) | 2024-12-12 |
Family
ID=91530268
Family Applications (1)
| Application Number | Title | Priority Date | Filing Date |
|---|---|---|---|
| PCT/US2024/032388 Ceased WO2024254055A1 (en) | 2023-06-09 | 2024-06-04 | Image segmentation and feature extraction using neural networks and a spatio-temporal filter |
Country Status (2)
| Country | Link |
|---|---|
| EP (1) | EP4724982A1 (en) |
| WO (1) | WO2024254055A1 (en) |
-
2024
- 2024-06-04 WO PCT/US2024/032388 patent/WO2024254055A1/en not_active Ceased
- 2024-06-04 EP EP24732816.4A patent/EP4724982A1/en active Pending
Non-Patent Citations (5)
| Title |
|---|
| EICHHARDT IVÁN ET AL: "Image-guided ToF depth upsampling: a survey", MACHINE VISION AND APPLICATIONS, SPRINGER VERLAG, DE, vol. 28, no. 3, 13 March 2017 (2017-03-13), pages 267 - 282, XP036217195, ISSN: 0932-8092, [retrieved on 20170313], DOI: 10.1007/S00138-017-0831-9 * |
| F. SULTANAA. SUFIANP. DUTTA: "''Evolution of image segmentation using deep convolutional neural network: a survey,", '' ARXIV PREPRINT ARXIV:2001.04074V3, 29 May 2020 (2020-05-29) |
| J. JOHNSONA. ALAHIL. FEI-FEI: "In Computer Vision-ECCV 2016: 14th European Conference, Amsterdam, The Netherlands", 27 March 2016, SPRINGER INTERNATIONAL PUBLISHING, article "Perceptual losses for real-time style transfer and super-resolution", pages: 694 - 711 |
| MOHSEN FAYYAZ ET AL: "STFCN: Spatio-Temporal FCN for Semantic Video Segmentation", 2 September 2016 (2016-09-02), XP055512490, Retrieved from the Internet <URL:https://arxiv.org/pdf/1608.05971.pdf> [retrieved on 20181004] * |
| SIMONYANZISSERMAN: "Very deep convolutional networks for large-scale image recognition", ARXIV PREPRINT ARXIV:1409.1556, 2014 |
Also Published As
| Publication number | Publication date |
|---|---|
| EP4724982A1 (en) | 2026-04-15 |
Similar Documents
| Publication | Publication Date | Title |
|---|---|---|
| Li et al. | Hyperspectral image super-resolution by band attention through adversarial learning | |
| Lee et al. | Mu-net: Multi-scale U-net for two-photon microscopy image denoising and restoration | |
| Engin et al. | Cycle-dehaze: Enhanced cyclegan for single image dehazing | |
| CN108256562A (en) | Well-marked target detection method and system based on Weakly supervised space-time cascade neural network | |
| Qu et al. | TransFuse: A unified transformer-based image fusion framework using self-supervised learning | |
| Shen et al. | Convolutional neural pyramid for image processing | |
| CN110866938B (en) | A fully automatic video moving object segmentation method | |
| Shi et al. | (SARN) spatial-wise attention residual network for image super-resolution | |
| Dixit et al. | A review of single image super resolution techniques using convolutional neural networks | |
| Zhang et al. | Irmamba: Pixel difference mamba with layer restoration for infrared small target detection | |
| Zhang et al. | Deep-learning-based point cloud completion methods: A review | |
| Liu et al. | MFID-Net: multi-scaled feature-fused image dehazing via dynamic weights | |
| Prasad et al. | Digital Image Enhancement using Conventional Neural Network | |
| Tomar et al. | Noise robust face super-resolution via learning of spatial attentive features | |
| Zhao et al. | Structure-texture dual preserving for remote sensing image super resolution | |
| Hou et al. | Multi-level features fusion via cross-layer guided attention for hyperspectral pansharpening | |
| Lai et al. | Video super-resolution via pre-frame constrained and deep-feature enhanced sparse reconstruction | |
| CN115187621A (en) | Automatic U-Net medical image contour extraction network integrating attention mechanism | |
| Mini et al. | A study in enhancement of satellite images | |
| Li et al. | LersGAN: A GAN-Based Model for Low-Light Remote Sensing Image Enhancement | |
| Jiang et al. | Refined Multi-Focus image fusion using Multi-Scale neural network with SpSwin Autoencoder-based matting | |
| US12561754B2 (en) | Method and system for processing image based on weighted multiple kernels | |
| Ramwala et al. | Reminiscent net: Conditional GAN-based old image de-creasing | |
| Su et al. | Attention-adaptive multi-scale feature aggregation dehazing network | |
| Samreen et al. | Enhanced Image Super Resolution Using ResNet Generative Adversarial Networks. |
Legal Events
| Date | Code | Title | Description |
|---|---|---|---|
| 121 | Ep: the epo has been informed by wipo that ep was designated in this application |
Ref document number: 24732816 Country of ref document: EP Kind code of ref document: A1 |
|
| WWE | Wipo information: entry into national phase |
Ref document number: 2024732816 Country of ref document: EP |
|
| NENP | Non-entry into the national phase |
Ref country code: DE |
|
| ENP | Entry into the national phase |
Ref document number: 2024732816 Country of ref document: EP Effective date: 20260109 |
|
| ENP | Entry into the national phase |
Ref document number: 2024732816 Country of ref document: EP Effective date: 20260109 |
|
| ENP | Entry into the national phase |
Ref document number: 2024732816 Country of ref document: EP Effective date: 20260109 |
|
| ENP | Entry into the national phase |
Ref document number: 2024732816 Country of ref document: EP Effective date: 20260109 |
|
| ENP | Entry into the national phase |
Ref document number: 2024732816 Country of ref document: EP Effective date: 20260109 |
|
| WWP | Wipo information: published in national office |
Ref document number: 2024732816 Country of ref document: EP |







