國立臺北科技大學資訊工程系計算機圖學實驗室
回頂端
系統(腳踏車)
系統(航照圖)
系統(人臉)
參考文獻
使用手冊
Weight map visualizer 使用手冊
Weight map visualizer是改造pytorch版本的yolov2而來
作者網址為https://github.com/marvis/pytorch-yolo2
目的是要將權重檔視覺化
所以必須先下載權重檔以及其設定檔
1. 到yolo作者的官網
Yolov2網址:https://pjreddie.com/darknet/yolov2/
2. 下載YOLOv2的權重檔yolov2.weights 大小為194MB 下載時間約1分30秒
3. 下載YOLOv2的設定檔yolov2.cfg
4. 進到
yolov2 github
5. 點選cfg資料夾
6. 找到並點擊yolov2.cfg
7. 在raw右鍵,選擇另存連結
8. 下載Weight map visualizer
網址為
https://github.com/jysh1214/yolov2-visualization
9. 解壓縮yolov2-visualization,並進入其資料夾
10. 將下載的yolov2.weights與yolov2.cfg檔放到cfg資料夾中
12. 下載python 網址:
https://www.python.org/downloads/
12. 安裝python
11. 開啟命令提示字元cmd,並將位置指向yolov2-visualization-master資料夾
12. 為了產生圖檔,需先在python中安裝opencv
13. 輸入指令 pip install opencv-python
14. 安裝深度學習的框架檔
15. 輸入指令 pip install torch torchvision
16. 安裝pytorch 網址:https://pytorch.org/get-started/locally/
15. 輸入指令 pip3 install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
17. 執行weight map visualizer
18. 輸入指令 python vis.py ./cfg/yolov2.cfg ./cfg/yolo.weights
19. 等待產生檔案,圖片會放在generated資料夾內