sync

cpp_old/msl_infer/Nmsl2.h

+#pragma once
+#include "Msl.h"
+
+class Nmsl2 : public Msl
+{
+public:
+	sptr<Resource> resRaw1;
+	sptr<Resource> resRaw2;
+	Net *fcNet1;
+	Net *fcNet2;
+	Net *catNet;
+	unsigned int batchSize;
+	unsigned int 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();
+
+};

cpp_old/msl_infer/Sampler.h

+#pragma once
+#include "../utils/common.h"
+
+class Sampler {
+public:
+    Sampler(glm::vec2 depthRange, unsigned int samples, bool outputRadius)
+        : _dispRange(1.0f / depthRange, samples), _outputRadius(outputRadius) {}
+
+    void sampleOnRays(sptr<CudaArray<float>> o_coords, sptr<CudaArray<float>> o_depths,
+                      sptr<CudaArray<glm::vec3>> rays, glm::vec3 rayCenter);
+
+private:
+    Range _dispRange;
+    bool _outputRadius;
+};
\ No newline at end of file