search

ardianumam / Tensorflow-TensorRT

This repository is for my YT video series about optimizing a Tensorflow deep learning model using TensorRT. We demonstrate optimizing LeNet-like model and YOLOv3 model, and get 3.7x and 1.5x faster for the former and the latter, respectively, compared to the original models.
303 stars 110 forks source link

How do you get the YOLOv3 Frozen Model used in this project #25

Open githubljj opened 5 years ago

githubljj commented 5 years ago

hi，this is great! i'm just wondering how do you train the YOLOv3 Frozen Model used in this project? thank U

yiwenwan2008 commented 4 years ago

did you figure out which yolov3 implementation the model used in this repository is using?

MuhammadAsadJaved commented 3 years ago

@githubljj @yiwenwan2008 I think it's TensorFlow-Yolov3 implementation. Train the model using Tensorflow and then convert .ckpt to .pb. You can use This project