Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 09:44:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Export pretrained alexnet

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-20 15:49:59 +02:00
parent ab10959f35
commit 0807a0f2b8
5 changed files with 63 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/alexnet/main.cpp
View File

@@ -4,6 +4,7 @@
#include <opencv2/opencv.hpp>
#include <model.hpp>
#include <conv2d.cuh>
std::vector<float> readAndNormalizeImage(const std::string& imagePath, int width, int height) {
// Read the image using OpenCV
@@ -28,6 +29,17 @@ std::vector<float> readAndNormalizeImage(const std::string& imagePath, int width
CUDANet::Model* createModel(const int inputSize, const int inputChannels, const int outputSize) {
CUDANet::Model *model =
new CUDANet::Model(inputSize, inputChannels, outputSize);
// AlexNet
CUDANet::Layers::Conv2d *conv1 = new CUDANet::Layers::Conv2d(
inputSize, inputChannels, 11, 4, 96, CUDANet::Layers::Padding::SAME, CUDANet::Layers::ActivationType::RELU
);
model->addLayer("conv1", conv1);
CUDANet::Layers::MaxPooling *pool1 = new CUDANet::Layers::MaxPooling(
3, 2
)
return model;
}