你可以使用如下步骤来使用两个训练模型:
导入所需的库,包括 Mask R-CNN 的预训练模型。
加载第一个模型:
model1 = modellib.MaskRCNN(mode="inference", config=config, model_dir=MODEL_DIR)
model1.load_weights(model1_path, by_name=True)
你可以使用如下步骤来使用两个训练模型:
导入所需的库,包括 Mask R-CNN 的预训练模型。
加载第一个模型:
model1 = modellib.MaskRCNN(mode="inference", config=config, model_dir=MODEL_DIR)
model1.load_weights(model1_path, by_name=True)