Как использовать мою собственную модель .pb на Jetson Nano вместо s sh -mobil enet -v2? - PullRequest
Новая
       117

Как использовать мою собственную модель .pb на Jetson Nano вместо s sh -mobil enet -v2?

1 голос
/ 18 марта 2020

ребята, у меня есть вопрос. Я могу запустить пример обнаружения объекта python на моем Jetson Nano. "s sh -mobil enet -v2" в этом случае. Это работает нормально ... Я также обучил свой собственный набор данных и есть файл .pb. Как я могу использовать его вместо "s sh -mobil enet -v2"? 

import jetson.inference
import jetson.utils

net = jetson.inference.detectNet("ssd-mobilenet-v2", threshold=0.5)
camera = jetson.utils.gstCamera(1280, 720, "0")  # RPIcam
display = jetson.utils.glDisplay()

while display.IsOpen():
    img, width, height = camera.CaptureRGBA()
    detections = net.Detect(img, width, height)
    display.RenderOnce(img, width, height)
    display.SetTitle("Object Detection | Network {:.0f} FPS".format(net.GetNetworkFPS()))

Спасибо

1 Ответ

0 голосов
/ 27 марта 2020

Файл .pb представляет собой замороженный график TensorFlow. Поэтому вам нужно установить TF и ​​зависимые от него библиотеки. Тем не менее, вы не можете сделать это для большинства библиотек, включая TF, используя pip install или sudo apt install, вам нужно собрать их из исходного кода или из их файла .whl.

Для более подробной информации следуйте пошаговым инструкциям Мадуиноса по установке TensorFlow, Numpy, Scipy, Pandas, OpenCV и др. c ...

...