作者ThePttUser (YhePttUser)
看板Python
標題[問題] YOLOv2怎麼跑單張圖 以及怎麼存出labels?
時間Fri Jun 12 19:17:37 2020
如題
我現在卡在我已經查過yolov2的help,我還是找不到可以只跑單張圖的參數,
另外幫圖上label這件事我不想另外存出圖,我想跑完直接給我單張圖的labels就好,
這些事情有辦法做到嗎?
另外跑圖有可能用gpu嗎? 我目前用的指令:
python flow --model cfg/yolo.cfg --load bin/yolov2.weights --imgdir
sample_img/
只用cpu跑,我覺得實在有點慢,但我後面單純加--gpu會錯誤。
我已經查了google一下午,都是關於v3的資訊,但因為我用windows,安裝v3對我來說
複雜很多,我認為v2已經堪用,因此想問問以上的問題。
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 123.192.90.125 (臺灣)
※ 文章網址: https://webptt.com/m.aspx?n=bbs/Python/M.1591960659.A.F9D.html
1F:推 andy086: 你要去改裡面的code,另外v3要安裝什麼?你沒有要訓練只 06/12 21:27
2F:→ andy086: 要拿他們的跑結果不是嗎? 06/12 21:27
3F:→ ThePttUser: 我看好像還要裝VS? 06/12 21:29
4F:→ ThePttUser: 那我再研究看看V3 06/12 21:30
5F:→ andy086: 不然你就改一下v2的code也行,找出他predict的那段改 06/12 21:46
6F:→ ilove0618: V3 如果你是使用darknet本身的框架的話你要安裝opencv 06/14 00:55
7F:→ ilove0618: 在電腦本身環境cuda cudnn就能使用gpu執行 如果要用ke 06/14 00:55
8F:→ ilove0618: ras或是pytorch的話還是得安裝cuda and cudbn 你可以 06/14 00:55
9F:→ ilove0618: 去看他的main看需要放上什麼參數可以執行gpu 我覺得v3 06/14 00:55
10F:→ ilove0618: 本身沒有很難安裝的 06/14 00:55
11F:→ MasterChang: 用opencv的dnn模塊會比較直接..... 06/24 00:24