[AI] yolov5: data download, image annotation, training, prediction, etc., super detailed notes; pytorch to onnx and then to TensorRT; training to recognize masks

Summary of AI Learning Catalog

1. Build yolov5 development environment

1.1 Official source code

https://gitcode.net/mirrors/ultralytics/yolov5/

1.2 Create a virtual environment

conda create --name yolov python=3.9 -y
conda activate yolov
conda install -n yolov  ipykernel
python -m ipykernel install --user --name yolov --display-name yolov

1.3 Installation

1) Install pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

2) Install cudnn

conda search cudnn
conda install cudnn==8.9.2.26

3) Download yolov5 source code:

git clone https://github.com/ultralytics/yolov5 

4) Install yolov5 dependency

cd yolov5
pip install -r requireme