Base class for segment five objects of Endoscopy Disease Detection and
Segmentation
database (EDD2020). Input is an image (cv:Mat).
Output is a struct of detection results, named MedicalSegmentationResult
.
Sample code :
Mat img = cv::imread("sample_medicalsegmentation.jpg");
auto medicalsegmentation =
vitis::ai::MedicalSegmentation::create
("FPN_Res18_Medical_segmentation",true);
auto results = medicalsegmentation->run(img);
// results is std::vector<cv::Mat>(5) for 5 classes.
// please check test samples for detail usage.
Quick Function Reference
The following table lists all the functions defined in the vitis::ai::MedicalSegmentation
class:
Type | Name | Arguments |
---|---|---|
std::unique_ptr< MedicalSegmentation > |
create |
|
vitis::ai::MedicalSegmentationResult
|
run |
|
std::vector< vitis::ai::MedicalSegmentationResult > |
run |
|
int | getInputWidth |
|
int | getInputHeight |
|
size_t | get_input_batch |
|