igor3357/

post-discharge-channel-segmentation

post-discharge-channel.pt is a YOLO26 segmentation model trained to identify post-discharge channels in shadowgraph images. The predicted masks can be used to measure channel area, dimensions, deformation, displacement, and lifetime in high-speed experimental recordings.

post_discharge_channel_segmentation_yolo26.py processes PNG and JPG frames in alphabetical order. It saves frames with segmentation masks, creates an MP4 video, calculates mask area versus time, exports the measurements to CSV, generates a plot with a moving-median trend, and creates a 4×4 grid of selected frames.

Place the model, script, and input folder together:

Install the dependencies and run the script:

All configurable parameters are located at the beginning of the script.

If you use this model, dataset, or accompanying software in your research, please cite one or more of the following publications, selecting those most relevant to your work: