You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
669 B
Python
23 lines
669 B
Python
5 years ago
|
from CompDatabase import ImageDB
|
||
|
from datetime import datetime
|
||
|
|
||
|
start_time = datetime.now()
|
||
|
# TODO save & load feature lists for the database, fix resized matching (injective resizing?)
|
||
|
|
||
|
root_folder = "Pictures"
|
||
|
import_folder = "import"
|
||
|
db_folder = "images"
|
||
|
samples = 200
|
||
|
dist_thresh = 80
|
||
|
match_thresh = 0.8
|
||
|
resize_dim = (500, 500)
|
||
|
threads = 4
|
||
|
|
||
|
img_db = ImageDB(root_folder, import_folder, db_folder, samples=samples, dist_thresh=dist_thresh,
|
||
|
match_thresh=match_thresh, resize_dim=resize_dim, threads=threads)
|
||
|
|
||
|
img_db.log("Starting work ...")
|
||
|
img_db.log("DB size: {0}".format(img_db.get_db_size()))
|
||
|
img_db.process_all_images()
|
||
|
img_db.log("Work done!")
|