Progressive and Aligned Pose Attention Transfer for Person Image Generation

Zhen Zhu, Tengteng Huang, Mengde Xu, Baoguang Shi, Wenqing Cheng, Xiang Bai

IEEE Trans. Pattern Anal. Mach. Intell 2022 · OpenReview

Versions

TL;DR

This paper introduces a progressive Generative Adversarial Network (GAN) framework for person pose transfer that breaks down a complex, global pose change into a sequence of simpler, local sub-steps. By leveraging novel Pose-Attentional and Aligned Pose-Attentional Transfer Blocks, the model generates highly realistic images with superior appearance and shape consistency while remaining remarkably lightweight compared to prior state-of-the-art architectures.


What Problem Does This Paper Address?

Imagine taking a single photo of a person standing sideways and trying to generate a realistic image of them facing forward, or mid-stride. This task—known as person pose transfer—is a major challenge in computer vision with applications ranging from virtual try-ons to video generation and data augmentation.

The core difficulty lies in handling large spatial deformations, managing occlusions (inferring body parts that weren't visible in the original image), and maintaining identity and texture consistency. Traditional generative models attempt this transformation in a single, massive step. The authors argue that this forces the network to navigate a highly complex global image manifold all at once, which frequently results in blurry details, warped clothing patterns, and anatomical distortions.


Methods

To tackle this complexity, the authors propose a Progressive Pose Attention Transfer Network. The intuition is simple: while a global pose change is difficult, breaking it down into a sequence of smaller, localized pose variations makes the problem much easier to solve.

The architecture consists of an Appearance Encoder, a Shape Encoder, a sequence of cascaded transfer blocks, and a Decoder:

Generator architecture of the proposed method featuring cascaded transfer blocks.

The paper introduces two distinct types of core transfer blocks:

  1. Pose-Attentional Transfer Block (PATB): This block implicitly models the relationship between the source and target poses. It uses the pose pathway to generate an element-wise attention mask, which is then multiplied directly with the appearance features.
  2. Aligned Pose-Attentional Transfer Block (APATB): Recognizing that PATB struggles to physically "move" complex textures across an image, the authors designed the APATB. It explicitly calculates a spatial correlation/alignment matrix between the source and target poses using a matrix inner product. This acts as a soft-warping mechanism, smoothly guiding feature patches to their correct target locations.
Detailed architectures of the PATB (left) and the APATB (right).

Datasets

The model was evaluated on two widely accepted benchmarks:

  • Market-1501: A challenging dataset containing low-resolution (128 × 64) images captured in real-world environments. It features highly complex, varied backgrounds and significant illumination changes.
  • DeepFashion (In-shop Clothes Retrieval Benchmark): A high-resolution (256 × 256) dataset featuring clean backgrounds and well-lit fashion imagery, making it ideal for evaluating how well a model preserves intricate clothing textures and details.

Key Insights and Ideas

  • Divide and Conquer the Manifold: Restricting pose and appearance variations to a localized perspective allows each block to operate on a simpler manifold structure, avoiding the pitfalls of global one-step transformations.
  • Explicit Alignment is Key for Textures: Implicit attention masks are great for feature selection but poor at spatial relocation. By explicitly computing pixel-to-pixel similarities between poses, the APATB can precisely map textures (like stripes or logos) from the source to the target coordinates.
  • Counter-Intuitive Parameter Efficiency: Despite having more convolutional layers, the Aligned Pose-Attentional Network (APATN) is significantly more parameter-saving than its predecessor (PATN). This is achieved by using $1 \times 1$ convolutions right after feature concatenation, which drastically reduces channel dimensions before deeper processing.

Key Results

  • State-of-the-Art Consistency: The APATN achieved the highest Structural Similarity (SSIM) scores on both datasets (0.326 on Market-1501 and 0.780 on DeepFashion), proving its superior ability to maintain appearance consistency.
  • Drastic Parameter Reduction: APATN requires only 16.93M parameters, compared to PATN's 41.36M and older architectures like Ma et al.'s, which ballooned to 437.09M.
  • Data Augmentation Powerhouse: The authors tested the network as a data generator for person re-identification (ReID) tasks. When real training data was severely restricted (to just 10%), augmenting the dataset with images generated by PATN boosted an Inception-v2 model's ReID accuracy from a dismal 5.7% to 42.3%.

Limitations and Future Work

Despite its strong performance, the authors note a few areas for improvement:

  • Background Blurring: Because the APATB focuses heavily on aligning foreground person features, it can sometimes confuse or blur complex backgrounds (particularly on the Market-1501 dataset).
  • Rare and Incomplete Poses: The model still struggles with highly unusual postures or heavily occluded inputs, occasionally failing to reconstruct fine details like hands or overlapping limbs.
  • Future Directions: The authors suggest that future iterations should explicitly separate the foreground and background generation processes, blending them at the final stage to preserve environmental clarity.

Loading related content...