安装torch torchvision :
pip install torch torchvision opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
永久设置镜像源(可选):
可以通过创建或编辑 ~/.pip/pip.conf
文件(在 Windows 上是 %USERPROFILE%\pip\pip.ini
),添加以下内容:
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
torchvision 中包含了ResNet-18 ,可以方便地加载、使用和修改这个模型,用于训练视频目标检测模型