public void IndexFiles(FileInfo[] imageFiles, BackgroundWorker IndexBgWorker, object argument = null)
{
BKTree <CEDDTreeNode> ceddtree = new BKTree <CEDDTreeNode>();
double[] ceddDiscriptor = null;
int totalFileCount = imageFiles.Length;
CEDD cedd = new CEDD();
for (int i = 0; i < totalFileCount; i++)
{
var fi = imageFiles[i];
using (Bitmap bmp = new Bitmap(Image.FromFile(fi.FullName)))
{
ceddDiscriptor = cedd.Apply(bmp);
}
CEDDTreeNode ceddTreeNode = new CEDDTreeNode
{
Id = i,
ImageName = fi.Name,
ImagePath = fi.FullName,
CEDDDiscriptor = ceddDiscriptor
};
ceddtree.add(ceddTreeNode);
IndexBgWorker.ReportProgress(i);
}
CEDDRepository <BKTree <CEDDTreeNode> > repo = new CEDDRepository <BKTree <CEDDTreeNode> >();
repo.Save(ceddtree);
CacheHelper.Remove("CeddIndexTree");
}
public List <ImageRecord> QueryImage(string queryImagePath, object argument = null)
{
List <ImageRecord> rtnImageList = new List <ImageRecord>();
CEDD_Descriptor.CEDD cedd = new CEDD_Descriptor.CEDD();
int goodMatchDistance = 35;
if (argument != null && argument is Int32)
{
goodMatchDistance = (int)argument;
}
double[] queryCeddDiscriptor;
using (Bitmap bmp = new Bitmap(Image.FromFile(queryImagePath)))
{
queryCeddDiscriptor = cedd.Apply(bmp);
}
Stopwatch sw = Stopwatch.StartNew();
BKTree <CEDDTreeNode> ceddTree = null;
if (!CacheHelper.Get <BKTree <CEDDTreeNode> >("CeddIndexTree", out ceddTree))
{
CEDDRepository <BKTree <CEDDTreeNode> > repo = new CEDDRepository <BKTree <CEDDTreeNode> >();
ceddTree = repo.Load();
if (ceddTree == null)
{
throw new InvalidOperationException("Please index CEDD with BK-Tree before querying the Image");
}
CacheHelper.Add <BKTree <CEDDTreeNode> >(ceddTree, "CeddIndexTree");
}
sw.Stop();
Debug.WriteLine("Load tooked {0} ms", sw.ElapsedMilliseconds);
CEDDTreeNode queryNode = new CEDDTreeNode
{
Id = 0,
ImagePath = queryImagePath,
CEDDDiscriptor = queryCeddDiscriptor
};
sw.Reset(); sw.Start();
Dictionary <CEDDTreeNode, Int32> result = ceddTree.query(queryNode, goodMatchDistance);
sw.Stop();
Debug.WriteLine("Query tooked {0} ms", sw.ElapsedMilliseconds);
foreach (KeyValuePair <CEDDTreeNode, Int32> ceddNode in result)
{
ImageRecord rec = new ImageRecord
{
Id = ceddNode.Key.Id,
ImageName = ceddNode.Key.ImageName,
ImagePath = ceddNode.Key.ImagePath,
Distance = ceddNode.Value
};
rtnImageList.Add(rec);
}
rtnImageList = rtnImageList.OrderBy(x => x.Distance).ToList();
return(rtnImageList);
}
