-
Notifications
You must be signed in to change notification settings - Fork 6
/
MyNeuralNetwork.h
37 lines (25 loc) · 753 Bytes
/
MyNeuralNetwork.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
// Fill out your copyright notice in the Description page of Project Settings.
#pragma once
#include "CoreMinimal.h"
#include "NeuralNetwork.h"
#include "PreOpenCVHeaders.h"
#include "OpenCVHelper.h"
#include <ThirdParty/OpenCV/include/opencv2/imgproc.hpp>
#include <ThirdParty/OpenCV/include/opencv2/highgui/highgui.hpp>
#include <ThirdParty/OpenCV/include/opencv2/core.hpp>
#include "PostOpenCVHeaders.h"
#include "MyNeuralNetwork.generated.h"
/**
*
*/
UCLASS()
class FPSTYLETRANSFER_API UMyNeuralNetwork : public UNeuralNetwork
{
GENERATED_BODY()
public:
UPROPERTY(Transient)
UNeuralNetwork* Network = nullptr;
UMyNeuralNetwork();
TArray<FColor> URunModel(cv::Mat image);
TArray<float> UPreProcessImage(cv::Mat image);
};