60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from yacs.config import CfgNode as CfgNode
|
|
_C = CfgNode()
|
|
cfg = _C
|
|
|
|
_C.RESIZE = 648
|
|
|
|
#Monitoring table of commodity identification System
|
|
_C.MONITORPATH = '../module/ieemoo-ai-searchv2/model/now/monitor.json'
|
|
|
|
_C.VIDEOPATH = '../module/ieemoo-ai-searchv2/videos'
|
|
|
|
#The number of results retrieved
|
|
_C.NUM_RESULT = 5
|
|
|
|
#RAFT img numbers
|
|
_C.NUM_RAFT = 10#6 #real nu: nu-1
|
|
|
|
#Path for storing video images
|
|
_C.TEST_IMG_DIR = '../module/ieemoo-ai-searchv2/imgs'
|
|
_C.ORIIMG = '../module/ieemoo-ai-searchv2/imgs'
|
|
|
|
_C.Tempimg = '../module/ieemoo-ai-searchv2/data'#扫码商品展示
|
|
_C.Tempvideos = '../module/ieemoo-ai-searchv2/tempvideos'#扫码商品展示
|
|
|
|
_C.DIM = 2048
|
|
_C.flag = False
|
|
|
|
#angle of rotation
|
|
_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-searchv2/model/now/model_best.pth'
|
|
|
|
#Weight of RAFT
|
|
_C.RAFTMODEL= '../module/ieemoo-ai-searchv2/model/now/raft-things.pth'
|
|
|
|
_C.DEVICE = 0
|
|
|
|
#Similarity threshold
|
|
_C.THRESHOLD = 0.9
|
|
|
|
#mask img
|
|
_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-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'
|
|
|
|
_C.Vre = 'http://api.test2.ieemoo.cn/emoo-api/intelligence/queryVideoCompareResult.do'
|
|
#_C.Vre = 'http://192.168.1.98:8088/emoo-api/intelligence/queryVideoCompareResult.do'
|
|
|
|
_C.Ocrimg = '../module/ieemoo-ai-ocr/imgs'#post ocr img
|
|
_C.Ocrtxt = '../module/ieemoo-ai-ocr/document'#post ocr txts
|
|
_C.Ocrvideo = '../module/ieemoo-ai-ocr/videos'#post ocr video
|