Skip to content

Commit

Permalink
Add instructions of minimial PyTorch version (#74)
Browse files Browse the repository at this point in the history
* Add instructions of minimial PyTorch version

* Minor fixes

* Move torch installation to the top
  • Loading branch information
zhiqwang authored Feb 26, 2021
1 parent 76f5a5d commit ecc0625
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,24 +31,32 @@ There are no extra compiled components in `yolort` and package dependencies are

### Installation and Inference Examples

- Above all, follow the [official instructions](https://pytorch.org/get-started/locally/) to install PyTorch 1.7.0+ and torchvision 0.8.1+

- Installation via Pip

Simple installation from PyPI
Simple installation from [PyPI](https://pypi.org/project/yolort/)

```bash
```shell
pip install -U yolort
```

Or from Source

```bash
```shell
# clone yolort repository locally
git clone https://github.com/zhiqwang/yolov5-rt-stack.git
cd yolov5-rt-stack
# install in editable mode
pip install -e .
```

- Install pycocotools (for evaluation on COCO):

```shell
pip install -U 'git+https://github.com/ppwwyyxx/cocoapi.git#subdirectory=PythonAPI'
```

- To read a source of image(s) and detect its objects 🔥

```python
Expand Down

0 comments on commit ecc0625

Please sign in to comment.