lichen/ieemoo-ai-searchv2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
李晨
2022-11-22 15:50:16 +08:00
parent 5e10fd86c1
commit 5e2c5fa39b
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
utils/config.py
View File

@ -5,9 +5,9 @@ cfg = _C
_C.RESIZE = 648 _C.RESIZE = 648
#Monitoring table of commodity identification System #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 #The number of results retrieved
_C.NUM_RESULT = 5 _C.NUM_RESULT = 5
@ -16,11 +16,11 @@ _C.NUM_RESULT = 5
_C.NUM_RAFT = 10#6 #real nu: nu-1 _C.NUM_RAFT = 10#6 #real nu: nu-1
#Path for storing video images #Path for storing video images
_C.TEST_IMG_DIR = '../module/ieemoo-ai-search/imgs' _C.TEST_IMG_DIR = '../module/ieemoo-ai-searchv2/imgs'
_C.ORIIMG = '../module/ieemoo-ai-search/imgs' _C.ORIIMG = '../module/ieemoo-ai-searchv2/imgs'
_C.Tempimg = '../module/ieemoo-ai-search/data'#扫码商品展示 _C.Tempimg = '../module/ieemoo-ai-searchv2/data'#扫码商品展示
_C.Tempvideos = '../module/ieemoo-ai-search/tempvideos'#扫码商品展示 _C.Tempvideos = '../module/ieemoo-ai-searchv2/tempvideos'#扫码商品展示
_C.DIM = 2048 _C.DIM = 2048
_C.flag = False _C.flag = False
@ -30,10 +30,10 @@ _C.ANGLES = [45,90,270,315] #45,90,135,180,225,270,315
#Weight of feature extraction #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' #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 #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 _C.DEVICE = 0
@ -41,12 +41,12 @@ _C.DEVICE = 0
_C.THRESHOLD = 0.9 _C.THRESHOLD = 0.9
#mask img #mask img
_C.MASKIMG = '../module/ieemoo-ai-search/model/now/masking.jpg' _C.MASKIMG = '../module/ieemoo-ai-searchv2/model/now/masking.jpg'
_C.MASKIMG_old = '../module/ieemoo-ai-search/model/now/masking_old.jpg' _C.MASKIMG_old = '../module/ieemoo-ai-searchv2/model/now/masking_old.jpg'
#fgbg mask img #fgbg mask img
_C.fgbgmask = '../module/ieemoo-ai-search/model/now/ori.jpg' _C.fgbgmask = '../module/ieemoo-ai-searchv2/model/now/ori.jpg'
_C.fgbgmask_old = '../module/ieemoo-ai-search/model/now/ori_old.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 = 'https://api.ieemoo.com/emoo-api/intelligence' #online
#_C.URL = 'http://api.test.ieemoo.com/emoo-api/intelligence' #_C.URL = 'http://api.test.ieemoo.com/emoo-api/intelligence'