This repository contains a YOLO26 model for automatic segmentation of shock waves in experimental shadowgraph images.
The model identifies visible shock-wave structures and returns instance segmentation masks. It is intended for automated processing of high-speed flow-visualization data in gas dynamics and shock-wave experiments.
Model authors: Igor Doroshchenko, Pavel Popov, Aleksei Moiseevskii, and Tahir Kuli-Zade.
A recent version of ultralytics with YOLO26 support is required. The accompanying script checks compatibility and can update ultralytics in the active Python environment when necessary.
Input Data
Place the model, script, and input directory in the same project folder:
The frames directory may contain PNG, JPG, or JPEG images. Files are sorted and processed alphabetically.
Running the Model
Open shock_wave_segmentation_yolo26.py in PyCharm, configure the parameters at the top of the script, and run it.
The main parameters include:
model filename;
input and output directories;
confidence and IoU thresholds;
inference image size and device selection;
mask transparency and contour settings;
polynomial approximation degree;
optional frame-time display;
first-frame time and time step between frames;
plot and legend settings for the extracted shock-front lines.
Output
The script creates three output directories.
results
Contains the original shadowgraph images with predicted segmentation masks.
results_with_lines
Contains the images with thin median lines representing the detected shock-wave fronts. Polynomial equations are not printed on these images.
For each mask, the script extracts a median shock-front line and automatically selects its representation as either:
y = f(x) for predominantly horizontal fronts;
x = f(y) for predominantly vertical fronts.
results_lines_on_grid
Contains separate plots of the extracted shock-front lines on a coordinate grid. The polynomial equation of each fitted line is displayed in the legend.
Pixel coordinates are used, with the origin located at the upper-left corner of the image by default.
Optional Frame Time
Frame time can be printed in the upper-left corner of the output images:
frame time = firstFrameTime + frame index × timeStep
The first frame after alphabetical sorting has index 0. The time unit is user-defined and must be consistent for firstFrameTime and timeStep.
Model Details
Task: Instance segmentation
Architecture: Ultralytics YOLO26
Target class: Shock wave
Input type: Experimental shadowgraph images
Output: Shock-wave instance masks
Weights:shock_seg_model_YOLO26.pt
Authors: Igor Doroshchenko, Pavel Popov, Aleksei Moiseevskii, and Tahir Kuli-Zade
Limitations
The model was developed for experimental shadowgraph images. Its performance may decrease for schlieren images, numerical visualizations, substantially different optical configurations, low-contrast shock fronts, strong illumination artifacts, or flow regimes not represented in the training dataset.
Predictions should be validated before the model is used for quantitative measurements. Spatial calibration is required to convert pixel coordinates into physical units.
Citation
If you use the exact model weights or the accompanying script, please cite this model repository:
bibtex
1@misc{doroshchenkoEtAl2026shockwaveyolo26,
2 author = {Doroshchenko, Igor and Popov, Pavel and Moiseevskii, Aleksei and Kuli-Zade, Tahir},
3 title = {YOLO26 Model for Shock-Wave Segmentation},
4 year = {2026},
5 publisher = {Hugging Face},
6 howpublished = {Hugging Face model repository},
7 url = {ADD_HUGGING_FACE_REPOSITORY_URL}
8}
Replace ADD_HUGGING_FACE_REPOSITORY_URL with the public URL of this repository after publication.
Related Publications
The publications below provide scientific and methodological background for computer-vision analysis of shock waves and other gas-dynamic structures in experimental shadowgraph images. They should not be interpreted as descriptions of the exact YOLO26 weights released in this repository.
If you use this model or the accompanying script in scientific work, please cite the model repository and one or both of the following publications, depending on the focus of your study.
Analysis of striped structures behind a shock wave
This article presents a computer-vision-based analysis of striped structures developing in the boundary layer behind a shock wave. It is particularly relevant to studies of shock-wave-induced unsteady flow structures and automated processing of high-speed flow-visualization data.
I. A. Znamenskaya, I. A. Doroshchenko, A. Kopylov, and T. A. Kuli-Zade, “Analysis of Striped Structures in the Boundary Layer Behind a Shock Wave Using Computer Vision Techniques,” Journal of Flow Visualization and Image Processing, vol. 33, no. 2, pp. 1–19, 2026. https://doi.org/10.1615/JFlowVisImageProc.2025058234
bibtex
1@article{znamenskaya2026stripedstructures,
2 author = {Znamenskaya, Irina A. and Doroshchenko, Igor A. and Kopylov, Anatoly and Kuli-Zade, Tahir A.},
3 title = {Analysis of Striped Structures in the Boundary Layer Behind a Shock Wave Using Computer Vision Techniques},
4 journal = {Journal of Flow Visualization and Image Processing},
5 year = {2026},
6 volume = {33},
7 number = {2},
8 pages = {1--19},
9 doi = {10.1615/JFlowVisImageProc.2025058234},
10 url = {https://doi.org/10.1615/JFlowVisImageProc.2025058234}
11}
Computer-vision analysis of experimental shadowgraph images
This article is the most directly relevant methodological reference for automatic analysis of shock waves and bow shocks in experimental shadowgraph images. It describes both classical computer-vision algorithms and YOLO-based deep learning for recognizing gas-dynamic structures.
I. A. Doroshchenko, “Analysis of the Experimental Flow Shadowgraph Images by Computer Vision Methods,” Numerical Methods and Programming (Vychislitel'nye Metody i Programmirovanie), vol. 24, no. 2, pp. 231–242, 2023. https://doi.org/10.26089/NumMet.v24r217
bibtex
1@article{doroshchenko2023shadowgraph,
2 author = {Doroshchenko, Igor A.},
3 title = {Analysis of the Experimental Flow Shadowgraph Images by Computer Vision Methods},
4 journal = {Numerical Methods and Programming (Vychislitel'nye Metody i Programmirovanie)},
5 year = {2023},
6 volume = {24},
7 number = {2},
8 pages = {231--242},
9 doi = {10.26089/NumMet.v24r217},
10 url = {https://doi.org/10.26089/NumMet.v24r217}
11}
For work focused specifically on the released YOLO26 weights, cite the model repository. For broader methodological and physical context, cite the most relevant publication above in addition to the repository.
