android-demo-app을 이용한 yolov5 객체탐지 앱 만들던 중 isTuple()INTERNAL ASSERT FAILED at 관한 에러가 발생했습니다. 어떤 해결방법이 있을까요?

이 줄(line)을 지우시고 여기서부터 질문을 작성해주세요 :slight_smile:

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
환경에서 작업중입니다.

로그만으론 뭐가 문제인지 잘 모르겠군요.
저도 한번 돌려보고 확인해 보도록 하겠습니다.

1개의 좋아요

이 글은 마지막 댓글이 달린지 오래(30일)되어 자동으로 닫혔습니다. 댓글 대신 새로운 글을 작성해주세요! :slight_smile: