Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement DRBN SKF #1353

Open
wants to merge 18 commits into
base: master
Choose a base branch
from
Open

Implement DRBN SKF #1353

wants to merge 18 commits into from

Conversation

Akasan
Copy link
Contributor

@Akasan Akasan commented Dec 23, 2023

#1190

のPRです

@Akasan Akasan requested a review from kyakuno December 23, 2023 05:30
@kyakuno
Copy link
Collaborator

kyakuno commented Jan 29, 2024

@kyakuno
Copy link
Collaborator

kyakuno commented Jan 29, 2024

@Akasan v2の画像が、結構、白飛びしているように見えるのですが、公式のpytorchでも同じような絵作りでしょうか?

@Akasan
Copy link
Contributor Author

Akasan commented Jan 29, 2024

@kyakuno
コメントありがとうございます。
該当データですぐに確認できそうになかったので、現在別データにて確認中です。

@Akasan
Copy link
Contributor Author

Akasan commented Jan 29, 2024

別のデータで確認してみましたが、確かに白飛びしているようです。
原因確認してみます

@Akasan
Copy link
Contributor Author

Akasan commented Feb 10, 2024

@kyakuno
現在、元リポジトリの実装との比較やONNX出力部分の確認など進めております。
ONNXにした後の出力の値が異なっているところまでは原因がわかったのでもう少し調査してみます

@Akasan
Copy link
Contributor Author

Akasan commented Feb 16, 2024

@kyakuno
調査を続けているのですが、精度が変わってしまう原因が現状掴めていません。
ONNX変換用で調整してたモデル定義の入出力が同じなところはまでは確認できていますが、ONNXファイルを使った推論をするときに出力結果が異なるようです。

経験としてONNXに変換したときに数値が変わってしまう経験がありましたら、なぜそのようになるのかや対応策などご教授いただけますと幸いです

私の方ではすぐに原因が掴めなさそうなので、調べつつも一旦コメントいただくまでは別で取り掛かっていたタスクを進められればと思います

@Akasan
Copy link
Contributor Author

Akasan commented Feb 18, 2024

@kyakuno
onnxruntimeを利用して推論すると元実装と同じ結果になることが確認できました。
ailia.Netで作ったときに結果が変わるようなのですが対処法はありますでしょうか?

@kyakuno
Copy link
Collaborator

kyakuno commented Feb 18, 2024

@Akasan onnx runtimeのほうで結果が一致するようでしたら、オプションで切り替えられるようにしていただければ、ailiaチームの方で解析させていただきます。

@Akasan
Copy link
Contributor Author

Akasan commented Feb 18, 2024

@Akasan onnx runtimeのほうで結果が一致するようでしたら、オプションで切り替えられるようにしていただければ、ailiaチームの方で解析させていただきます。

onnxruntimeの方では結果が一致しましたので、--onnxオプションを指定すればonnxruntimeで稼働するように変更しておきましたので、ご確認いただければと思います

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants