Yolox 예제 노트북 코드 관련으로 문의 드립니다

제가 이해한 걸 정리하면

16라인을 주석 처리한 후에
model = onnx.load_model(“yolox_l.onnx”)
model = optimize_model(model)

====================================================

editor = ModelEditor(model)
input_tensor_name = get_pure_input_names(model)[0] #<<==입력텐서이름찾기

editor.convert_input_type(input_tensor_name, TensorType.UINT8)

====================================================

== 안에 든
위 코드를 이용해서 모델의 입력 타입을 float에서 int8로 바꾸어야한다는 말씀이군요.
답변 감사합니다.

1 Like