[問題] YOLOv2怎麼跑單張圖 以及怎麼存出labels?

[ThePttUser (YhePttUser)]

如題

我現在卡在我已經查過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://www.ptt.cc/bbs/Python/M.1591960659.A.F9D.html
推 andy086: 你要去改裡面的code，另外v3要安裝什麼？你沒有要訓練只
→ andy086: 要拿他們的跑結果不是嗎？
→ ThePttUser: 我看好像還要裝VS?
→ ThePttUser: 那我再研究看看V3
→ andy086: 不然你就改一下v2的code也行，找出他predict的那段改
→ ilove0618: V3 如果你是使用darknet本身的框架的話你要安裝opencv
→ ilove0618: 在電腦本身環境cuda cudnn就能使用gpu執行 如果要用ke
→ ilove0618: ras或是pytorch的話還是得安裝cuda and cudbn 你可以
→ ilove0618: 去看他的main看需要放上什麼參數可以執行gpu 我覺得v3
→ ilove0618: 本身沒有很難安裝的
→ MasterChang: 用opencv的dnn模塊會比較直接.....