Add Alexnet README

This commit is contained in:
2024-04-22 21:59:08 +02:00
parent 9a9b034ce5
commit 157d4c736b
2 changed files with 53 additions and 19 deletions

View File

@@ -0,0 +1,28 @@
# AlexNet
AlexNet Inference on CUDANet
## Usage
1. Export pytorch AlexNet weight pretrained on ImageNet (requires pytorch and torchvision):
```sh
python alexnet.py
```
2. Follow the instructions from repository root to build CUDANet library.
3. Build AlexNet
```sh
mkdir build
cd build
cmake -S ..
make
```
4. Run AlexNet inference
```sh
alexnet ../alexnet_weights.bin ../image.jpg
```