USEFULL/qw-gallery-scenery
1
0
Fork 0
mirror of https://github.com/qwertyforce/scenery.git synced 2025-05-31 11:42:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update calculate_color_hist.ts

Browse Source
This commit is contained in:
qwertyforce 2020-09-22 20:28:22 +03:00
parent a673eaa17b
commit 50c285354c
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/bulk_calculate_color_hist_and_similarities/calculate_color_hist.ts
View File

@ -25,6 +25,10 @@ const PATH_TO_IMAGES = path.join("..", "..", "..", 'public', 'images')
async function calculate_color_hist(){ async function calculate_color_hist(){
const images = await db_ops.image_ops.get_all_images() const images = await db_ops.image_ops.get_all_images()
for(const image of images){ for(const image of images){
const check_if_already_calculated= await db_ops.image_search.get_color_hist_by_id(image.id)
if(check_if_already_calculated.length!==0){
continue
}
console.log(image.id) console.log(image.id)
const img = await cv.imreadAsync(`${PATH_TO_IMAGES}/${image.id}.${image.file_ext}`); const img = await cv.imreadAsync(`${PATH_TO_IMAGES}/${image.id}.${image.file_ext}`);
let rgb_hist = await cv.calcHistAsync(img, histAxes) let rgb_hist = await cv.calcHistAsync(img, histAxes)
@ -42,6 +46,10 @@ async function calc_similarities() {
const get_all_hist=await db_ops.image_search.get_all_color_hists() const get_all_hist=await db_ops.image_search.get_all_color_hists()
console.time(); console.time();
for (let i = 0; i < images.length - 1; i++) { for (let i = 0; i < images.length - 1; i++) {
const check_if_already_calculated=await db_ops.image_search.get_color_similarities_by_id(images[i].id)
if(check_if_already_calculated.length!==0){
continue
}
const target_image = get_all_hist.find((el)=>el.id===images[i].id) const target_image = get_all_hist.find((el)=>el.id===images[i].id)
const target_hist = new cv.Mat(target_image.color_hist, cv.CV_32F) const target_hist = new cv.Mat(target_image.color_hist, cv.CV_32F)
const similarities=[] const similarities=[]
@ -56,10 +64,10 @@ async function calc_similarities() {
await db_ops.image_search.add_color_similarities_by_id(images[i].id,similarities) await db_ops.image_search.add_color_similarities_by_id(images[i].id,similarities)
} }
console.timeEnd() console.timeEnd()
process.exit()
} }
async function calc_color_hists_and_similarities() { async function calc_color_hists_and_similarities() {
await calculate_color_hist() await calculate_color_hist()
await calc_similarities() await calc_similarities()
process.exit()
} }
calc_color_hists_and_similarities() calc_color_hists_and_similarities()