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;
}