kazuhito00/animeganv2-onnx-sample

「PyTorch Implementation of AnimeGANv2」のPythonでのONNX推論サンプル

Github Trophy

About Me

• 名古屋で組込みエンジニア(C/C++)をしていました。
  I was an embedded engineer(C/C++) in Japan Aichi Nagoya.
• 最近は画像処理(C++/Python)、Web(Azure/AWS/SFDC)やIoTの仕事もしています。
  Recently I've been working on image processing(C++/Python), Web Application(Azure/AWS/SFDC) and IoT.
• 機械学習やKaggle勉強中(🥉×２)です。
  Currently studying machine learning and Kaggle(🥉×2)
• 物体検出やセマンティックセグメンテーションに興味があります。
  I am interested in ObjectDetection and SemanticSegmentation

Works

リポジトリ一覧、寄稿等は「WORKS.md」を参照ください。
※画像を多く使用しており通信量が多くなる懸念があるため、モバイル回線や通信料従量制の方はご注意ください
　また、サムネイルの画像が小さいため、PCでの閲覧を推奨しています
Please refer to "WORKS.md" for the list of repositories and contributions.
Because many images are used and there is a concern that the amount of communication will increase,
please be careful if you are using a mobile line or a pay-as-you-go communication system.
Also, since the thumbnail image is small, we recommend viewing it on a PC.

Gitstar Ranking

「Gitstar Ranking(Kazuhito00)」にてスター数の多いリポジトリが確認できます。
You can check repositories with many stars on "Gitstar Ranking(Kazuhito00)".

Recent Blog Posts

• Apr 20 - MeshFlowで動画のブレ補正をお試し🦔
• Apr 18 - Keras Code examples 「Keypoint Detection with Transfer Learning」のダメそうなポイントのメモ🦔
• Apr 13 - GlobalFlowNetで動画のブレ補正をお試し🦔
• Apr 12 - VidStabで動画のブレ補正🐦
• Apr 11 - MobileSAM と EfficientSAM も ONNX推論お試し👀

How to reach me

何かご相談の際には、TwitterのDMから連絡をお願いします。
If you have any questions, please contact us via Twitter DM.

SNS

Technologies

animeganv2-onnx-sample's Issues

runtimeerror

runs fine in colab but get this error running on a huggingface space

Traceback (most recent call last):
File "app.py", line 7, in
onnx_session = onnxruntime.InferenceSession("/home/user/app/face_paint_512_v2_0.onnx")
File "/home/user/.local/lib/python3.8/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 324, in init
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "/home/user/.local/lib/python3.8/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 358, in _create_inference_session
sess = C.InferenceSession(session_options, self._model_path, True, self._read_config_from_model)
RuntimeError: /onnxruntime_src/onnxruntime/core/platform/posix/env.cc:173 onnxruntime::{anonymous}::PosixThread::PosixThread(const char*, int, unsigned int ()(int, Eigen::ThreadPoolInterface), Eigen::ThreadPoolInterface*, const onnxruntime::ThreadOptions&) pthread_setaffinity_np failed, error code: 0 error msg:

resize aspect ratio

currently the image resizing is making the images squished if it is not 1:1 already