guillherms/stride-architecture-components-v1

STRIDE Architecture Threat Modeling Dataset (AWS & Azure)

📌 Overview

This dataset was created to enable automatic STRIDE threat modeling from cloud architecture diagrams (AWS and Azure).

The goal is to detect architectural components in diagrams and support automated threat identification based on data flows and trust boundaries.

Annotations were created using Label Studio in YOLO format.

Total images: 4190 Total classes: 32

🎯 Purpose

• —Detect cloud architecture components
• —Identify trust boundaries
• —Enable automated STRIDE threat analysis
• —Support security research in diagram-based threat modeling

🖼 Image Source

Images were collected from publicly available architecture diagrams on the internet, including:

• —AWS reference architectures
• —Azure reference architectures
• —Cloud solution blog posts
• —Technical documentation examples

🔄 Data Augmentation Strategy

To improve robustness and simulate real-world variations, the following augmentations were applied:

• —_BW (Black and White conversion)
• —_sharp (Sharpen filter)
• —_contrast (Contrast adjustment)
• —_gamma_hi (High gamma correction)
• —_gamma_lo (Low gamma correction)
• —_jpeg50 (JPEG compression 50%)
• —_blur1 (Gaussian blur level 1)
• —_noise6 (Noise injection level 6)
• —_degrade80 (Quality degradation 80%)

🏷 Classes (32 total)

📦 Dataset Format

YOLO format:

<classid> <xcenter> <y_center> <width> <height>

Values are normalized between 0 and 1.

🏗 Structure

train/images, val/images, test/images train/labels, val/labels, test/labels data.yaml

👤 Author

Guilherme Santos Vision Architecture Analyzer – 2026