#pragma once #include "Msl.h" class Nmsl2 : public Msl { public: sptr resRaw1; sptr 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(); };