diff --git a/README.md b/README.md index b883612..05f4de4 100644 --- a/README.md +++ b/README.md @@ -39,6 +39,7 @@ pip3 install -r requirements.txt ``` Execute this command using `sudo` if you are not using a virtual environment. + ## How to Use You need to use a TensorFlow neural network on which running the inferences. We provide in this repository a pre-trained model taken from [the TensorFlow Model Zoo](https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md). There you can find more models ready to work. You can choose a different one, but make sure that it outputs __boxes__ (masks are not supported yet), and specify its dataset and the path to the `.pb` file containing the graph in `objectdetector.yml`. diff --git a/requirements.txt b/requirements.txt index 175d80f..bf9bcc0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ -numpy==1.14 # to avoid deprecation warnings -tensorflow==1.14.0 +numpy~=1.19.2 +tensorflow==2.4.1 opencv-python Pillow cprint \ No newline at end of file