English | 简体中文
The following contents are related to interactive video annotation in EISeg, mainly including model selection, data preparation and instructions.
The VTK package should be additionally installed for 3D visualization, please try the following:
pip install vtk
Interactive video object segmentation is based on EISeg interactive segmentation algorithms and MIVOS algorithm. It is an efficient image and video annotation software based on PaddlePaddle.
EISeg 1.0 covers high-quality interactive video object segmentation models in different directions such as general, liver, CT spinal structures, etc., which is convenient for developers to quickly annotate videos and reduce the cost of annotation. For 3D Medical Imaging Annotation, we regard medical slice data as the video frames, and realize the labeling of 3D medical images by using video annotation method.
Before using EISeg, please download the propagation model parameters. If you want to use the 3D display function, you can check the 3D display function in the Display
menu.
Model Type | Applicable Scenarios | Model Architecture | Download Link | Corresponding Propagation Model Download Link |
---|---|---|---|---|
High Performance Model | Image annotation in generic scenarios | HRNet18_OCR64 | static_hrnet18_ocr64_cocolvis | static_propagation |
Lightweight Model | Image annotation in generic scenarios | HRNet18s_OCR48 | static_hrnet18s_ocr48_cocolvis | static_propagation |
High Performance Model | Image annotation in generic scenarios | EdgeFlow | static_edgeflow_cocolvis | static_propagation |
High Performance Model | Annotation in portrait scenarios | HRNet18_OCR64 | static_hrnet18_ocr64_human | static_propagation |
Lightweight Model | Annotation in portrait scenarios | HRNet18s_OCR48 | static_hrnet18s_ocr48_human | static_propagation |
Lightweight Model | Annotation of liver in medical scenarios | HRNet18s_OCR48 | static_hrnet18s_ocr48_lits | static_propagation_lits |
Lightweight Model | Annotation of CT Spinal Structures in medical scenarios | HRNet18s_OCR48 | static_hrnet18s_ocr48_MRSpineSeg | static_propagation_spine |
- Due to the large computation in video segmentation, it is recommended to use the compyter with a graphics card, and the number of labeled video should not exceed 100 frames. If the video exceeds the number of frames, you can use cut_video.py to cut video.
- 3D medical imaging annotation is based on interactive video segmentation, so please convert the sliced images into mp4 format before labeling. The script is: medical2video.py.
After opening the software, make the following settings before annotating:
-
Load Model Parameter
Select the appropriate network and load the corresponding model parameters. After downloading and decompressing the right model and parameters, the model structure
*.pdmodel
and the corresponding model parameters*.pdiparams
should be put into the same directory, and only the location of the model parameter at the end of*.pdiparams
need to be selected when loading the model. The initialization of the static model takes some time, please wait patiently until the model is loaded. The correctly loaded model parameters will be recorded inRecent Model Parameters
, which can be easily switched, and the exiting model parameter will be loaded automatically the next time you open the software. -
Load Image
Open the image or image folder. Things go well when you see that the main screen image is loaded correctly and the image path is rightly shown in
Data List
. -
Add/Load Label
Add/load labels. New labels can be created by
Add Label
, which are divided into 4 columns corresponding to pixel value, description, color and deletion. The newly created labels can be saved as txt files bySave Label List
, and other collaborators can import labels byLoad Label List
. Labels imported by loading will be loaded automatically after restarting the software. -
Load Propagation Model Parameter
Select the corresponding propagation model parameters. After downloading and decompressing the right model and parameters, the model structure
*.pdmodel
and the corresponding model parameters*.pdiparams
should be put into the same directory, and only one of the location of the model parameter at the end of*.pdiparams
need to be selected when loading the model. The other models will be loaded automatically. -
Annotate Referance frame
During interactive annotation, users add positive and negative points with left and right mouse clicks, respectively. After finishing interactive segmentation, you can push Space button and the tool generates a polygon frame around the target border. Users can adjust the polygon vertexes to further improve segmentation accuracy.
-
Autosave
You can choose the right folder and have the
autosave
set up, so that the annotated image will be saved automatically when switching images.