diff --git a/enums/ModelTypeEnum.py b/enums/ModelTypeEnum.py index 8168344..6f619f7 100644 --- a/enums/ModelTypeEnum.py +++ b/enums/ModelTypeEnum.py @@ -131,8 +131,7 @@ class ModelType(Enum): 'seg_nclass': 3, 'segRegionCnt': 2, 'segPar': { - #'modelSize': (640, 360), - 'modelSize': (1920, 1080), + 'modelSize': (640, 360), 'mean': (0.485, 0.456, 0.406), 'std': (0.229, 0.224, 0.225), 'predResize': True, @@ -141,8 +140,7 @@ class ModelType(Enum): 'mixFunction': { 'function': tracfficAccidentMixFunction, 'pars': { - #'modelSize': (640, 360), - 'modelSize': (1920,1080), + 'modelSize': (640, 360), 'RoadArea': 16000, 'roadVehicleAngle': 15, 'speedRoadVehicleAngleMax': 75, @@ -694,12 +692,13 @@ class ModelType(Enum): 'function': tracfficAccidentMixFunction, 'pars': { 'modelSize': (640, 360), - #'modelSize': (1920,1080), 'RoadArea': 16000, 'roadVehicleAngle': 15, 'speedRoadVehicleAngleMax': 75, 'roundness': 1.0, 'cls': 10, + 'CarId':1, + 'CthcId':1, 'vehicleFactor': 0.1, 'confThres': 0.25, 'roadIou': 0.6,