#ifndef REMAIN_FILTER_H
#define REMAIN_FILTER_H

#include <torch/script.h>
#include <opencv2/opencv.hpp>
#include <fstream>
#include <string>
#include <iomanip>
#include <stdlib.h>
#include <vector>
#include <iostream>
#include "nvidia_gpu_util.h"
#include "log_util.h"

struct RemainFilter
{
private:
    bool is_gpu;
    torch::jit::script::Module module;
    int gpu_id;

public:
    static RemainFilter* instance;
    static RemainFilter* getInstance();

public:
    RemainFilter();
    ~RemainFilter();
    RemainFilter(int gpu_id);
    float whichRemain(unsigned char *pBuf);

};


#endif