mirror of
https://github.com/azaion/detections.git
synced 2026-06-23 12:21:07 +00:00
This commit is contained in:
@@ -1,5 +1,6 @@
|
|||||||
import os
|
import os
|
||||||
import tempfile
|
import tempfile
|
||||||
|
cimport constants_inf
|
||||||
from engines.tensorrt_engine cimport TensorRTEngine
|
from engines.tensorrt_engine cimport TensorRTEngine
|
||||||
from loader_http_client cimport LoaderHttpClient, LoadResult
|
from loader_http_client cimport LoaderHttpClient, LoadResult
|
||||||
|
|
||||||
@@ -21,7 +22,6 @@ cdef class JetsonTensorRTEngine(TensorRTEngine):
|
|||||||
@staticmethod
|
@staticmethod
|
||||||
def _download_calib_cache(LoaderHttpClient loader_client, str models_dir):
|
def _download_calib_cache(LoaderHttpClient loader_client, str models_dir):
|
||||||
cdef LoadResult res
|
cdef LoadResult res
|
||||||
import constants_inf
|
|
||||||
try:
|
try:
|
||||||
res = loader_client.load_big_small_resource(
|
res = loader_client.load_big_small_resource(
|
||||||
constants_inf.INT8_CALIB_CACHE_FILE, models_dir
|
constants_inf.INT8_CALIB_CACHE_FILE, models_dir
|
||||||
|
|||||||
Reference in New Issue
Block a user