I have trained a model using Tensorflow 2 Object Detector API on a remote server. The server has the following setup:
- Tensorflow 2.3.1
- CUDA 10.1
- cuDNN 7.6
- Python 3.6.8
- Ubuntu 16.04
- Docker 18.09.2
The model I used for my project is the (EfficientDet D4 1024x1024). The trained model is working well, so then I exported the model using the
exporter_main_v2.py from Tensorflow.
Now I would like to use the exported model into my Jetson TX2 for inference. For this I have pulled the docker image with the following setup:
- Tensorflow 2.3.1+nv20.11
- CUDA 10.2
- cuDNN NA
- Python 3.6.9
- Ubuntu 16.04 LTS
- Docker 18.09.2
For the inference on the Jetson TX2, I used this guide Tensorflow 2 Object Detector API. In the aforemetioned guide, I realized that I needed to install the Tensorflow 2 Object Detector API to perform inference on the Jetson TX2. The commands I executed in the docker container to install the Tensorflow 2 Object Detector API were almost the same as the remote server (Except from the protoc):
curl -OL https://github.com/protocolbuffers/protobuf/releases/download/v3.17.3/protoc-3.17.3-linux-aarch_64.zip unzip protoc-3.17.3-linux-aarch_64.zip -d /usr/local bin/protoc) unzip protoc-3.17.3-linux-aarch_64.zip -d /usr/local include/* rm -f protoc-3.17.3-linux-aarch_64.zip git clone https://github.com/tensorflow/models.git) cd models/research protoc object_detection/protos/*.proto --python_out=. cp object_detection/packages/tf2/setup.py . python3 -m pip install --use-feature=2020-resolver .
However, when I executed the last command
python3 -m pip install --use-feature=2020-resolver .. It gives me the following error:
ERROR: Could not find a version that satisfies the requirement tensorflow>=2.5.0 (from tf-models-official) (from versions: none) ERROR: No matching distribution found for tensorflow>=2.5.0
According to the pip Tensorflow-addons. There is no support for the hardware architecture of the JetsonTX2 (aarch64) on linux, and that might be causing the error.
My Questions are:
- Is there a way get rid off the above error?
- Is there any guide on how to install Tensorflow 2 Object Detector API on the Jetson TX2?
- Is it necessary to install Tensorflow 2 Object Detector API on the Jetson TX2 to perform inference?