Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1f708f3ea395387cb5137d3c0fde0819a30e5a1b
CUDANet/examples/alexnet
History
LordMathis 1592f06121 Add opencv to alexnet usage
2024-04-22 22:56:29 +02:00
..
alexnet.py
Fix missing predict argument
2024-04-22 22:04:46 +02:00
CMakeLists.txt
Start implementing AlexNet
2024-04-18 22:31:14 +02:00
main.cpp
Add Alexnet README
2024-04-22 21:59:08 +02:00
README.md
Add opencv to alexnet usage
2024-04-22 22:56:29 +02:00

README.md

AlexNet

AlexNet Inference on CUDANet

Usage

  1. Export pytorch AlexNet weight pretrained on ImageNet (requires pytorch and torchvision):
python alexnet.py

  1. Follow the instructions from repository root to build CUDANet library.

  2. Build AlexNet (requires OpenCV for image loading and normalization)

mkdir build
cd build
cmake -S ..
make
  1. Run AlexNet inference
alexnet ../alexnet_weights.bin ../image.jpg