search

Qidian213 / deep_sort_yolov3

Real-time Multi-person tracker using YOLO v3 and deep_sort with tensorflow
GNU General Public License v3.0
1.66k stars 591 forks source link

How to I create file .pb tensorflow ? #97

Open NguyenThaiHoc1 opened 5 years ago

NguyenThaiHoc1 commented 5 years ago

I saw source using is mars-small128.pb But convert.py to .h5 ?

I hope people expland to me this problem and How can i do ?

Thanks everyone for reading

kazuya-n commented 5 years ago

mars-small128.pb is trained on MARS dataset. and used by deepsort to solve assignment problem. On the other hand, convert.py genarates *.h5. It's YOLO weights.

If you want to train deepsort and generate original .pb file, check this repo

zengdiqing1994 commented 5 years ago

mars-small128.pb is trained on MARS dataset. and used by deepsort to solve assignment problem. On the other hand, convert.py genarates *.h5. It's YOLO weights.

If you want to train deepsort and generate original .pb file, check this repo

Hello, if I want to train my own car dataset, how to use cos_metric_learning repo you said to train? Should I set my car dataset to the same format of market1501 reid dataset?

Thanks for your reply~