Skip to content

salehahr/graph-extraction-datagen

Repository files navigation

graph-training

Generation of training data for the graph extraction of endoscopic images.

Workflow

Video Workflow

  1. Append new video data to end of video_data.py
  2. Run before_filter.py
  3. In MATLAB, run filtering/Bladder_vessels.m (filepath can be input manually)
  4. Run after_filter.py

Image Folder Workflow

  1. In video_data.py:
    1. Update/Set variable video_filepath = "C:/My/FullPath/ImageFolder" to folder containing the .png images
    2. Set variable use_images = True
    3. If you want to use the FFT filter, set variable fft_filter = True
  2. Run before_filter.py
  3. In MATLAB, filtering/Bladder_vessels.m:
    1. Update/Set variable line 30 VIDEO_FILEPATH = 'C:/My/FullPath/ImageFolder'
    2. Run filtering/Bladder_vessels.m
  4. Run after_filter.py
Functions Description
before_filter.py Extracts and crops video frames
filtering/Bladder_vessels.m Applies B-COSFIRE filter to cropped images
after_filter.py Applies: mask, thresholding, skeletonising, graph generation

Folders

Folder Description
raw Raw video stills
cropped Cropped images, 256x256px
filtered Filtered images
masked Filtered images masked with a circular mask
threshed Thresholded images
skeleton Skeletonised images
graphs Graphs saved as .json files
overlay Graph overlaid on cropped image

About

Generates training data for graph extraction.

Topics

Resources

Stars

Watchers

Forks

Contributors 2

  •  
  •