Update card to 512 + 3D mesh demo + TRT 10.16 benchmarks
Browse files- infer_trt.py +1 -1
infer_trt.py
CHANGED
|
@@ -34,7 +34,7 @@ except ImportError as e: # pragma: no cove
|
|
| 34 |
HERE = Path(__file__).resolve().parent
|
| 35 |
ONNX = HERE / "embedl_sam3dbody_int8.onnx"
|
| 36 |
ENGINE = HERE / "embedl_sam3dbody_int8.engine"
|
| 37 |
-
INPUT_SIZE =
|
| 38 |
IMAGENET_MEAN = np.array([0.485, 0.456, 0.406], dtype=np.float32)
|
| 39 |
IMAGENET_STD = np.array([0.229, 0.224, 0.225], dtype=np.float32)
|
| 40 |
LOG = trt.Logger(trt.Logger.WARNING)
|
|
|
|
| 34 |
HERE = Path(__file__).resolve().parent
|
| 35 |
ONNX = HERE / "embedl_sam3dbody_int8.onnx"
|
| 36 |
ENGINE = HERE / "embedl_sam3dbody_int8.engine"
|
| 37 |
+
INPUT_SIZE = 512
|
| 38 |
IMAGENET_MEAN = np.array([0.485, 0.456, 0.406], dtype=np.float32)
|
| 39 |
IMAGENET_STD = np.array([0.229, 0.224, 0.225], dtype=np.float32)
|
| 40 |
LOG = trt.Logger(trt.Logger.WARNING)
|