diff --git a/utils/config.py b/utils/config.py index 92f52b1..e409a8a 100644 --- a/utils/config.py +++ b/utils/config.py @@ -5,9 +5,9 @@ cfg = _C _C.RESIZE = 648 #Monitoring table of commodity identification System -_C.MONITORPATH = '../module/ieemoo-ai-search/model/now/monitor.json' +_C.MONITORPATH = '../module/ieemoo-ai-searchv2/model/now/monitor.json' -_C.VIDEOPATH = '../module/ieemoo-ai-search/videos' +_C.VIDEOPATH = '../module/ieemoo-ai-searchv2/videos' #The number of results retrieved _C.NUM_RESULT = 5 @@ -16,11 +16,11 @@ _C.NUM_RESULT = 5 _C.NUM_RAFT = 10#6 #real nu: nu-1 #Path for storing video images -_C.TEST_IMG_DIR = '../module/ieemoo-ai-search/imgs' -_C.ORIIMG = '../module/ieemoo-ai-search/imgs' +_C.TEST_IMG_DIR = '../module/ieemoo-ai-searchv2/imgs' +_C.ORIIMG = '../module/ieemoo-ai-searchv2/imgs' -_C.Tempimg = '../module/ieemoo-ai-search/data'#扫码商品展示 -_C.Tempvideos = '../module/ieemoo-ai-search/tempvideos'#扫码商品展示 +_C.Tempimg = '../module/ieemoo-ai-searchv2/data'#扫码商品展示 +_C.Tempvideos = '../module/ieemoo-ai-searchv2/tempvideos'#扫码商品展示 _C.DIM = 2048 _C.flag = False @@ -30,10 +30,10 @@ _C.ANGLES = [45,90,270,315] #45,90,135,180,225,270,315 #Weight of feature extraction #_C.NETWORK = '../../module/ieemoo-ai-search/model/now/model_best.pth' #retrieval_feature 测试 -_C.NETWORK = '../module/ieemoo-ai-search/model/now/model_best.pth' +_C.NETWORK = '../module/ieemoo-ai-searchv2/model/now/model_best.pth' #Weight of RAFT -_C.RAFTMODEL= '../module/ieemoo-ai-search/model/now/raft-things.pth' +_C.RAFTMODEL= '../module/ieemoo-ai-searchv2/model/now/raft-things.pth' _C.DEVICE = 0 @@ -41,12 +41,12 @@ _C.DEVICE = 0 _C.THRESHOLD = 0.9 #mask img -_C.MASKIMG = '../module/ieemoo-ai-search/model/now/masking.jpg' -_C.MASKIMG_old = '../module/ieemoo-ai-search/model/now/masking_old.jpg' +_C.MASKIMG = '../module/ieemoo-ai-searchv2/model/now/masking.jpg' +_C.MASKIMG_old = '../module/ieemoo-ai-searchv2/model/now/masking_old.jpg' #fgbg mask img -_C.fgbgmask = '../module/ieemoo-ai-search/model/now/ori.jpg' -_C.fgbgmask_old = '../module/ieemoo-ai-search/model/now/ori_old.jpg' +_C.fgbgmask = '../module/ieemoo-ai-searchv2/model/now/ori.jpg' +_C.fgbgmask_old = '../module/ieemoo-ai-searchv2/model/now/ori_old.jpg' _C.URL = 'https://api.ieemoo.com/emoo-api/intelligence' #online #_C.URL = 'http://api.test.ieemoo.com/emoo-api/intelligence'