Nmsl2.h 448 Bytes
Newer Older
Nianchen Deng's avatar
Nianchen Deng committed
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
#pragma once
#include "Msl.h"

class Nmsl2 : public Msl
{
public:
	sptr<Resource> resRaw1;
	sptr<Resource> resRaw2;
	Net *fcNet1;
	Net *fcNet2;
	Net *catNet;
	uint batchSize;
	uint samples;

	Nmsl2(int batchSize, int samples);

	virtual bool load(const std::string &netDir);

	virtual void bindResources(Resource *resEncoded, Resource *resDepths, Resource *resColors);

	virtual bool infer();

	virtual void dispose();

};