mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
28 lines
394 B
Markdown
28 lines
394 B
Markdown
# 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
|
|
``` |