audio_wm/extract.h
2025-02-10 23:13:41 +08:00

31 lines
764 B
C++

#pragma once
#ifndef EXTRACT_H
#define EXTRACT_H
#include <vector>
#include <iostream>
#include <cmath>
#include <fstream>
#include <sndfile.h>
#include <opencv2/opencv.hpp>
#include <fftw3.h>
#include <string>
using namespace std;
int wm_extract(const wchar_t* path, int* wm);
vector<double> PQIMfft_extract(vector<vector<double>> &Gp_w, vector<int> &b, int M, double DD, double EE);
double drzh(const vector<vector<int>>& ww, vector<vector<int>>& W1, int pp, int qq);
double psnrzh(const vector<double>& yw, const vector<double>& yo);
double getWsr(const vector<double>& yw, const vector<double>& yo);
vector<vector<int>> igeneral(vector<vector<int>> W1, int m, int n, int a, int b);
// std::vector<int> randperm(int length, unsigned int seed);
#endif