Как загрузить модель без функции cv2.dnn.readNetFromCaffe? - PullRequest
2 голосов
/ 04 апреля 2020

Я загружаю код этого учебника и запускаю его. Я пытался изменить модель. Но я не могу понять, как изменить эту строку:

net = cv2.dnn.readNetFromCaffe(args["prototxt"], args["model"])

Согласно документу . readNetFromCaffe() читает модель сети, хранящуюся в формате Caffe Framework. Но моя модель - EffcientDet, она не в формате фреймворка Caffe. Также функция возвращает net объект. Можно ли использовать мою модель без readNetFromCaffe()

Я изменил эти строки

ap.add_argument("-p", "--prototxt", required=True,
    help="path to Caffe 'deploy' prototxt file")
ap.add_argument("-m", "--model", required=True,
    help="path to Caffe pre-trained model")
net = cv2.dnn.readNetFromCaffe(args["prototxt"], args["model"])

на

#ap.add_argument("-p", "--prototxt", required=True,
#   help="path to Caffe 'deploy' prototxt file")
ap.add_argument("-m", "--model", required=True,
    help="path to Caffe pre-trained model")
net = (args["model"])

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

...