Экспорт весов Mask_rcnn и подача новых изображений - PullRequest
0 голосов
/ 11 февраля 2020

Я не знаю, будет ли мой вопрос иметь смысл, так как я новичок в ML и, в частности, сегментация экземпляров.

Недавно я следил за какой-то записной книжкой, чтобы обучить модель Mask_rcnn (версия для портпорта) с моей собственной базой данных. Все прошло нормально. Однако мне хотелось бы знать, можно ли будет экспортировать модель, чтобы потом добавить к ней любое изображение. На данный момент кажется, что можно попробовать только с изображениями, которые уже есть в базе данных. Кроме того, при вызове modellib.load_image_gt () всегда необходимо использовать test_set.

image_id = 32 image, image_meta, gt_class_id, gt_bbox, gt_mask = modellib.load_image_gt (test_set, config,

)
image_id, use_mini_mask=False)
info = test_set.image_info[image_id]
print("image ID: {}.{} ({}) {}".format(info["source"], info["id"], image_id, 
test_set.image_reference(image_id)))

Моя цель состоит в том, чтобы обучить модель выполнять сегментацию экземпляров и использовать ее позже, чтобы обнаруживать некоторые объекты и записывать их положение на изображении, но изображения должны быть в отдельном файле, чем обучающий / набор данных проверки.

Спасибо за вашу помощь!

...