Sahi:用于大型图像/小物体切片推理的视觉库

2021-01-31 14:01:14

到目前为止,对象检测和实例分割是Computer Vision中最重要的应用程序领域。 但是,在实际使用中,小物体的检测和大图像的推断仍然是主要问题。 SAHI可以帮助开发人员克服这些现实问题。 结果= get_sliced_prediction(图像,检测模型,slice_height = 256,slice_width = 256,overlap_height_ratio = 0.2,overlap_width_ratio = 0.2) sahi库当前仅支持MMDetection模型。 但是,添加新框架很容易。 您需要做的就是在model.py中创建一个实现DetectionModel类的新类。 您可以将MMDetection包装器作为参考。