Msl.cpp 611 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
26
27
28
#include "Msl.h"
#include <time.h>

Msl::Msl() : net(nullptr) {}

bool Msl::load(const std::string &netPath) {
    net = new Net();
    if (net->load(netPath))
        return true;
    dispose();
    return false;
}

void Msl::bindResources(Resource *resEncoded, Resource *resDepths, Resource *resColors) {
    net->bindResource("Encoded", resEncoded);
    net->bindResource("Depths", resDepths);
    net->bindResource("Colors", resColors);
}

bool Msl::infer() { return net->infer(); }

void Msl::dispose() {
    if (net != nullptr) {
        net->dispose();
        delete net;
        net = nullptr;
    }
}