mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 02:04:26 +00:00
Add Alexnet README
This commit is contained in:
28
examples/alexnet/README.md
Normal file
28
examples/alexnet/README.md
Normal 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
|
||||
```
|
||||
Reference in New Issue
Block a user