product quantization IVFADC算法在Windows下的实现

% load the vectors from test/training/query sets, and the groundtruth % if dataset='random', a random dataset is generated on the fly % ='siftsmall', the siftsmall is used (from basedir directory) % ='sift', the sift dataset is used (frome basedir directory) % ='gist', the sift dataset is used (frome basedir directory) dataset = 'sift'; pq_test_load_vectors; %---[ Search parameters ]--- k = 10000; % number of elements to be returned nsq = 8; % number of subquantizers to be used (m in the paper) k_prime = 1024; % Learn the Coarse quantizer code structure t0 = cputime; cq = cq_new (k_prime, vtrain); cq_vtrain_idx = cq_assign (vtrain,cq); residual_train = residual_vector (vtrain,cq,cq_vtrain_idx); tcqlearn = cputime - t0; % Learn the PQ code structure t0 = cputime; pq = pq_new (nsq, residual_train); tpqlearn = cputime - t0; % encode the database vectors t0 = cputime; cq_vbase_idx = cq_assign (vbase,cq); ivf_cq_base = ivf_pq_assign(vbase,cq,k_prime,pq); tpqencode = cputime - t0; %---encode the vquery vectors--- % t0 = cputime; % pq_vquery = pq_vquery_assign(pq,vquery); % tpqqueryencode = cputime - t0; %---[ perform the search and compare with the ground-truth ]--- t0 = cputime; %[ids_pqc, dis_pqc] = pq_search (pq, cbase, pq_vquery, k); %SDC %query_residual = residual_vector (vquery,cq,cq_vbase_idx); %ADC for w =[1,16] [idx, dis] = yael_nn (cq, vquery, w, 2); [ids_pqc, dis_pqc] = ivf_pq_search (pq, ivf_cq_base, vquery, k,idx,cq); %ADC %[ids_pqc, dis_pqc] = pq_search (pq, cbase, vquery, k); %ADC % compute search statistics pq_test_compute_stats end % fprintf ('SDC learn = %.3f s\n', tpqlearn); % fprintf ('SDC encode = %.3f s\n', tpqencode); % fprintf ('SDC encodequery = %.3f s\n', tpqqueryencode); % fprintf ('SDC search = %.3f s for %d query vectors in a database of %d vectors\n', tpq, nquery, nbase); %load('share.mat')