python - 是否可以仅使用 .pb 文件评估预训练模型？

Question

我已经下载了这个源。

https://github.com/Austin-Ellsworth/Vehicle_Recognition_System/find/main

用于车辆检测 + 品牌/型号识别。

品牌/模型识别部分使用 TensorFlow/MobileNetsV3 完成。Weights.pb、Labels、Config 文件可用。

但没有检查点文件。而且也不知道他们用来训练模型的数据集。

那么是否可以仅使用这些文件评估此分类器模型？（例如获取地图？...）

如果没有，使用不同的数据集评估它会给我一个准确的地图吗？

最有效的方法是什么？

谢谢

标签： pythontensorflowmobilenet

Answer 1

*.pb文件代表 protobuf。tensorflow 使用 Protobuf 来保存模型图和权重。

模型实际上是一个移植到在原始 YOLOv3 权重上运行的 tensorflow 的 YOLOv3。它是在 COCO 数据集上训练的，所以它会给你在这个数据集上类似的准确性。