37 lines
962 B
Python
37 lines
962 B
Python
import os, sys
|
|
from SIFT import SIFT
|
|
from retrieval import retrieval
|
|
from pprint import pprint
|
|
|
|
|
|
def main():
|
|
print("start")
|
|
# img = cv.imread("beach.jpg")
|
|
# cv.imshow("Image", img)
|
|
# from matplotlib import pyplot as plt
|
|
# plt.hist(img.ravel(),10,[0,256]); plt.show()
|
|
# gray_img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
|
|
# cv.imshow("Gray Image", gray_img)
|
|
# cv.waitKey()
|
|
|
|
# for para_test in [8, 16]:
|
|
# print("test using " + str(para_test))
|
|
# assert retrieval("2") == "2"
|
|
# print("test using " + str(para_test) + " passed")
|
|
|
|
# flower
|
|
# assert retrieval("6") == "7" # 731, horse
|
|
# assert retrieval("1") == "1" # 126, 138, beach
|
|
# assert retrieval("4") == "4" # 458, dinosaur
|
|
# assert retrieval("5") == "6" # 6xx means flower category
|
|
#
|
|
assert retrieval("2") == "2" # 2xx means building
|
|
#
|
|
# assert retrieval('3') == "3"
|
|
|
|
print("test done")
|
|
sys.exit()
|
|
|
|
|
|
main()
|