88 lines
2.7 KiB
Python
88 lines
2.7 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.VreTest = 'http://api.test.ieemoo.cn/emoo-api/intelligence/queryVideoCompareResult.do' #test
|
|
_C.VreTest2 = 'http://api.test2.ieemoo.cn/emoo-api/intelligence/queryVideoCompareResult.do' #test2
|
|
_C.Vre = 'https://api.ieemoo.com/emoo-api/intelligence/queryVideoCompareResult.do' #online
|
|
|
|
_C.backups = '../module/output/videos'
|
|
_C.riskControl = 'http://api.test2.ieemoo.com/emoo-api/riskControl/shoporder/checkUploadAiImages?storeId=32001001'
|
|
|
|
_C.Ocrimg = '../module/ieemoo-ai-assist/imgs'#post ocr img
|
|
_C.Ocrtxt = '../module/ieemoo-ai-assist/document'#post ocr txts
|
|
_C.Ocrvideo = '../module/ieemoo-ai-assist/videos'#post ocr video
|
|
|
|
#obs
|
|
_C.obs_access_key_id='LHXJC7GIC2NNUUHHTNVI'
|
|
_C.obs_secret_access_key='sVWvEItrFKWPp5DxeMvX8jLFU69iXPpzkjuMX3iM'
|
|
_C.obs_server='https://obs.cn-east-3.myhuaweicloud.com'
|
|
_C.obs_bucketName = 'ieemoo-ai'
|
|
|
|
#cos
|
|
_C.cos_secret_id = 'AKIDIVcVFnBMMLCqbFF6lPr6930tc8VlrlS0'
|
|
_C.cos_secret_key = 'oYGrJG2d3LTUPBoFTZVEmC1kQ1ObdTkr'
|
|
_C.cos_region = 'ap-shanghai'
|
|
_C.cos_Bucket = 'ieemoo-ai-1321281601'
|
|
|
|
#s3
|
|
_C.s3_secret_id = 'v1wAfxqZqJYbA7kuB9dc'
|
|
_C.s3_secret_key = 'ZICGRJetjNIXhnqH1yfraoigydWoF2WTggtO9AhR'
|
|
_C.s3_region = '172.16.0.186:9000'
|
|
_C.s3_Bucket = 'ieemoo-ai'
|
|
|
|
#cos or obs
|
|
_C.obs = True
|
|
_C.cos = False
|
|
_C.s3 = True
|
|
|