Script to create new bat file with a retrained TensorFlow pb file
Hello there,
We are using this library in our project, and it is performing faster than TensorFlow Lite. Great job on speed up Mobilenet with MPSCNN!
The problem we are facing is, we are not using the pre-trained ImageNet 1000 classes, instead, we retrained Mobilenet with our own dataset. In order to take benefits from your code, we will have to create a new bat file.
Do you have any scripts or instructions to create the bat file? I understand that we will need to extract weights and biases from TensorFlow file, and also we need to merge the parameters into single file, but I am new to MPSCNN and TensorFlow, so a script should be very helpful.
Also, I think we will need to update MobileNetsData.swift to fit the new bat file, is there anything else we need to change? - We are using Mobilenet_v1_1.0_224.
Thanks!