Msl.h 310 Bytes
Newer Older
Nianchen Deng's avatar
Nianchen Deng committed
1
#pragma once
Nianchen Deng's avatar
sync    
Nianchen Deng committed
2
#include "../utils/common.h"
Nianchen Deng's avatar
Nianchen Deng committed
3
4
#include "Net.h"

Nianchen Deng's avatar
sync    
Nianchen Deng committed
5
class Msl {
Nianchen Deng's avatar
Nianchen Deng committed
6
7
8
public:
    Net *net;

Nianchen Deng's avatar
sync    
Nianchen Deng committed
9
    Msl();
Nianchen Deng's avatar
Nianchen Deng committed
10

Nianchen Deng's avatar
sync    
Nianchen Deng committed
11
12
13
14
    virtual bool load(const std::string &netDir);
    virtual void bindResources(Resource *resEncoded, Resource *resDepths, Resource *resColors);
    virtual bool infer();
    virtual void dispose();
Nianchen Deng's avatar
Nianchen Deng committed
15
};