17 lines
509 B
Python
17 lines
509 B
Python
from loguru import logger
|
|
from DrGraph.util.LogUtils import init_log
|
|
from GPUtil import getAvailable, getGPUs
|
|
from DrGraph.appIOs.conf.ModelUtils import *
|
|
from DrGraph.Bussiness.Bussiness import BussinessBase
|
|
|
|
if __name__ == '__main__':
|
|
from os.path import dirname, realpath
|
|
base_dir: str = dirname(realpath(__file__))
|
|
init_log(base_dir, 'algDev')
|
|
|
|
busi = 'illParking'
|
|
opt={'gpu':'3090','business':busi}
|
|
aiAlg = BussinessBase.createModel(opt)
|
|
if aiAlg:
|
|
aiAlg.run()
|
|
|