yolov5 install 추가
This commit is contained in:
85
README.md
85
README.md
@@ -1,36 +1,49 @@
|
||||
# MG_ENGINE_SIM
|
||||
|
||||
위험성 평가기반 작업자 안전관리 및 감독 협조 기술 개발 (시뮬레이터)
|
||||
|
||||
### System Requirements
|
||||
- ubuntu 20.04
|
||||
- python 3.8
|
||||
- python packages : requirements.txt
|
||||
|
||||
### Pytorch, Torchvision install
|
||||
- conda(CUDA 11.7 기준)
|
||||
```
|
||||
conda install pytorch torchvision pytorch-cuda=11.7 -c pytorch -c nvidia
|
||||
```
|
||||
- pip
|
||||
```
|
||||
pip3 install torch torchvision torchaudio
|
||||
```
|
||||
- ARM architecture
|
||||
|
||||
JetPack 5.0.2(L4T R35.1.0) 기준
|
||||
1. pytorch(version 1.12)
|
||||
```
|
||||
wget https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
|
||||
apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
|
||||
pip3 install Cython
|
||||
pip3 install torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
|
||||
```
|
||||
2. torchvision(version 0.13)
|
||||
```
|
||||
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
|
||||
git clone --branch release/0.13 https://github.com/pytorch/vision torchvision
|
||||
cd torchvision
|
||||
export BUILD_VERSION=0.13.0
|
||||
python3 setup.py install --user
|
||||
```
|
||||
# MG_ENGINE_SIM
|
||||
|
||||
위험성 평가기반 작업자 안전관리 및 감독 협조 기술 개발 (시뮬레이터)
|
||||
|
||||
### System Requirements
|
||||
- ubuntu 20.04
|
||||
- python 3.8
|
||||
- python packages : requirements.txt
|
||||
|
||||
### Pytorch, Torchvision install
|
||||
- conda(CUDA 11.7 기준)
|
||||
```
|
||||
conda install pytorch torchvision pytorch-cuda=11.7 -c pytorch -c nvidia
|
||||
```
|
||||
- pip
|
||||
```
|
||||
pip3 install torch torchvision torchaudio
|
||||
```
|
||||
- ARM architecture
|
||||
|
||||
JetPack 5.0.2(L4T R35.1.0) 기준
|
||||
1. pytorch(version 1.12)
|
||||
```
|
||||
wget https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
|
||||
apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
|
||||
pip3 install Cython
|
||||
pip3 install torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
|
||||
```
|
||||
2. torchvision(version 0.13)
|
||||
```
|
||||
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
|
||||
git clone --branch release/0.13 https://github.com/pytorch/vision torchvision
|
||||
cd torchvision
|
||||
export BUILD_VERSION=0.13.0
|
||||
python3 setup.py install --user
|
||||
```
|
||||
|
||||
### yolov5 install
|
||||
DL/OD 폴더에 저장
|
||||
```
|
||||
cd DL
|
||||
git clone https://github.com/ultralytics/yolov5.git OD
|
||||
```
|
||||
|
||||
git commit id : 23c492321290266810e08fa5ee9a23fc9d6a571f
|
||||
|
||||
```
|
||||
git reset hard -- 23c492321290266810e08fa5ee9a23fc9d6a571f
|
||||
```
|
||||
Reference in New Issue
Block a user