姑苏区网站建设,自己怎么开发棋牌软件,企业门户网站建站,网站开发要花费多少钱本范例演示使用非常有名的目标检测框架detectron2 🤗🤗
在自己的数据集(balloon数据)上训练实例分割模型MaskRCNN的方法。
detectron2框架的设计有以下一些优点: 1,强大:提供了包括目标检测、实例分割、全景分割等非常广泛的视觉任务模型库。 2,灵活:可以通过注册机…本范例演示使用非常有名的目标检测框架detectron2 🤗🤗
在自己的数据集(balloon数据)上训练实例分割模型MaskRCNN的方法。
detectron2框架的设计有以下一些优点: 1,强大:提供了包括目标检测、实例分割、全景分割等非常广泛的视觉任务模型库。 2,灵活:可以通过注册机制自定义模块或模型结构,从而进行扩展和改进。 3,易用:通过list of dict格式定义自己的数据集, 简单好用。 我们首先需要安装并导入detectron库~
!pipinstall'git+https://github.com/facebookresearch/detectron2.git'
!pipinstalltorchkerasimportnumpyasnp
importos,json,cv2,random
fromPILimportImageimporttorchimportdetectron2
fromdetectron2.configimportget_cfg
fromdetectron2.utils.visualizerimportVisualizer
fromdetectron2.dataimportMetadataCatalog,DatasetCatalog
fromdetectron2importmodel_zoo
fromdetectron2.engineimportDefaultPredictor#fromdetectron2.utils.loggerimportsetup_logger
#setup_logger()defcv2_show(arr):
img=Image.fromarray(cv2.cvtCo