update,

vinniesniper-54816/task1/_lab/001-test-integration/draft1.py

@@ -0,0 +1,36 @@
+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()