如何从github上克隆库、跑库

1,497 阅读2分钟

第一步:在Github上找到想要的库,以YOLOv3项目为例。

1655211392961.jpg

第二步:拷贝这个库到自己的电脑上,下载到本地。

方法一:在GitHub上,Code -> Download ZIP(有的时候会有一些问题,不建议)

1655211543820.png

方法二:使用Git

GitHub中项目的README.md部分仔细阅读(说明书),其中:

1655211735457.jpg

复制Git命令git clone https://github.com/BobLiu20/YOLOv3_PyTorch.git在cmd中运行,此处将下载的代码放在桌面上,默认已经安装了Git:

(Git的下载安装及配置参考:zhuanlan.zhihu.com/p/123195804)

1655212005051.jpg

这样就可以下载到本地桌面:

1655212101198.jpg

第三步:按照说明书上的指示,运行代码需要安装一些第三方库

作者写在了README.md部分:

1655212548456.png

打开桌面下载的YOLOv3_PyTorch代码包中requirements.txt文件可以看见跑这个项目需要用到的库:

1655212715955.jpg

一个一个用pip安装非常繁琐,复制上述第三行代码pip3 install -r requirements.txt --user在cmd中运行即可自动安装:

1655212916591.jpg

第四步:下载数据集

1655213028537.jpg

同样在cmd中运行:

1655213114279.jpg

第五步:下载预训练权重

根据README.md,需要下载预训练权重。YOLOv3作为一个深度卷积神经网络,整体分为两部分,backbone和其余部分。backbone里面的参数初始化不是随机初始化的,而是加载预训练的数据,即提前在其他数据集上训练过,让模型有了一个基本的识图能力,接下来模型的其余部分再放到coco数据集上去训练,使得它在具有识图能力的基础上获得检测物体的能力,所以需要一个预训练的权重文件用来初始化backbone参数。

image.png

该文件darknet53_weights_pytorch.pth是一个pth文件,即训练的权重文件。

谷歌云盘或者百度云盘即可下载,下载好了之后把它放到项目的weights文件夹中。

第六步:训练模型

根据README.md,需要到training文件夹里启动Python脚本training.py,同时training.py需要一系列外部参数,这些外部参数定义和赋值放在 params.py中:

1655214051954.jpg

在cmd中运行上述代码:

image.png

你以为这样就可以顺利的训练了吗?no!肯定会报错。

其实在复现代码没有想象的那么简单,把代码、数据下载下来,然后回车,程序就运行了?没有这么简单,都要先debug一遍。

之后就是跟着报错问题一步一步改,直到程序成功运行~

其实复现GitHub中的代码就只要跟着README.md一步一步来就好了,that's all.