The simple program of Faster R-CNN with pytorch and torchvision.
 |
 |
Make the anaconda environment.
This code is also available for CPU environment.
PyTorch: https://pytorch.org/get-started/locally/
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
You have to change some part of this line.
conda install numpy
pip install opencv-python
OpenCV conda installation is not recommended.