이 줄(line)을 지우시고 여기서부터 질문을 작성해주세요
android-demo-app을 이용한 yolov5 객체탐지 앱 만들던 중
Caused by: com.facebook.jni.CppException: isTuple()INTERNAL ASSERT FAILED at "../aten/src/ATen/core/ivalue_inl.h":1908, please report a bug to PyTorch. Expected Tuple but got String () Exception raised from toTuple at ../aten/src/ATen/core/ivalue_inl.h:1908 (most recent call first):
이러한 오류가 발생했습니다.
GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite 여기서 클론하여 train.py를 이용해 학습시켜 얻은 best.pt 파일을
expert.py를 이용해 torchscript를 변환 후
import torch
from torch.utils.mobile_optimizer import optimize_for_mobile
torchscript_model = "/content/drive/MyDrive/v5sbest.torchscript"
export_model_name = "v5s.torchscript.ptl"
model = torch.jit.load(torchscript_model)
model.eval()
optimized_model = optimize_for_mobile(model).cuda()
optimized_model._save_for_lite_interpreter(export_model_name)
print(f"mobile optimized model exported to {export_model_name}")
해당 코드를 이용해 ptl파일로 변환해주었습니다. 그리고 얻은 ptl파일을 안드로이드 스튜디오를 이용해 작성중인 코드에 넣으니 저러한 오류가 발생했습니다. 해결할 수 있는 좋은 방법이 있을까요?
저는 코랩을 이용하여 작성중이며
python 3.8.0
torch 1.12.1+cu113
환경에서 작업중입니다.