Nmsl2.h 464 Bytes
Newer Older
Nianchen Deng's avatar
Nianchen Deng committed
1
2
3
4
5
6
7
8
9
10
11
#pragma once
#include "Msl.h"

class Nmsl2 : public Msl
{
public:
	sptr<Resource> resRaw1;
	sptr<Resource> resRaw2;
	Net *fcNet1;
	Net *fcNet2;
	Net *catNet;
Nianchen Deng's avatar
Nianchen Deng committed
12
13
	unsigned int batchSize;
	unsigned int samples;
Nianchen Deng's avatar
Nianchen Deng committed
14
15
16
17
18
19
20
21
22

	Nmsl2(int batchSize, int samples);

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

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

	virtual bool infer();

Nianchen Deng's avatar
sync    
Nianchen Deng committed
23
	virtual void dispose();
Nianchen Deng's avatar
Nianchen Deng committed
24
25

};