- Pytorchで学習した重みを、YOLOのFPGA実装のPSで読み込むCのヘッダファイルに変換します。
- Pytorchを用いて学習
- 重みファイル (
.pt)のパスとクラス数を設定して、$ make
- 結果が
./final_output下に配置されます。
-
Notifications
You must be signed in to change notification settings - Fork 1
yarakigit/design_contest_yolo_change_ps_to_pl
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Converts pytorch yolo format weights to C header files for bare-metal (FPGA implementation)
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published