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
#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'