1. 程式人生 > >Faster RCNN修改demo.py檔案實現圖片的批量測試與儲存

Faster RCNN修改demo.py檔案實現圖片的批量測試與儲存

關於Faster R-CNN Tensorflow+python 3.5 在Windows10環境下配置實現,可以參看這裡。執行在demo.py檔案中測試資料中原始碼設定僅檢測幾張圖片供參考,原始的程式碼段如下。

im_names = ['000456.jpg', '000457.jpg', '000542.jpg', '001150.jpg',
                '001763.jpg', '004545.jpg']
    for im_name in im_names:
        print('~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~')
        print('Demo for data/demo/{}'.format(im_name))
        demo(sess, net, im_name)

    plt.show()

問題1:一旦測試圖片多起來,桌面上一下子顯示出十幾個figure視窗看起來有點小刺激,所以可以考慮修改以下程式碼,進行批量測試和儲存。

主要修改下檔案的儲存路徑與輸出格式,因為不少測試圖片經plt.savefig函式儲存,如果不加其他引數設定會導致影象上存在不少空白位置,plt.savefig函式的詳細引數設定可以參考官方文件:

 im_names = os.listdir(cfg.FLAGS2["data_dir"]+'/demo') #測試圖片所在位置
    for im_name in im_names:
        print('~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~')
        print('Demo for data/demo/{}'.format(im_name))
        demo(sess, net, im_name)
        #儲存測試圖片所在位置,並設定輸出格式
        plt.savefig(cfg.FLAGS2["data_dir"]+'/test_result/'+ im_name, format = 'png',transparent = True,pad_inches = 0,dpi = 300,bbox_inches = 'tight') 

   # plt.show()

問題2:就是demo.py檔案在對同一張圖片上每個視窗 僅顯示某一類圖片,而不是在每張影象上顯示所有類別

相關推薦

Faster RCNN修改demo.py檔案實現圖片批量測試儲存

關於Faster R-CNN Tensorflow+python 3.5 在Windows10環境下配置實現,可以參看這裡。執行在demo.py檔案中測試資料中原始碼設定僅檢測幾張圖片供參考,原始的程式碼段如下。 im_names = ['000456.jpg', '000

py-faster-rcnn 修改demo.py檔案將所有目標框都顯示在同一張圖裡

#!/usr/bin/env python # -------------------------------------------------------- # Faster R-CNN # Copyright (c) 2015 Microsoft # Licensed

py-faster-rcnndemo.py程式碼C++版本的程式碼對比: part01 鋪墊, demo.py引入的模組

★ python程式碼 tools/demo.py 中import的內容, 是整個程式碼流程的鋪墊, 理解了import的內容, 對理解後續的python程式碼和C++ 程式碼都有幫助. demo.py的import內容: import _

Python之使用unittest框架和HTMLTestRunner.py檔案實現多個測試用例執行測試報告生成

HTMLTestRunner.py檔案下載連結:http://tungwaiyip.info/software/HTMLTestRunner.html 一直想讓使用HTMLTestRunner生成的測試報告可以儲存在指定的檔案目錄下,並以當前執行的時間進行命名,在網上有找

目標檢測框架py-faster-rcnn修改anchor_box

算法 框架 ams const fas nbsp 這一 res weight 眾所周知,anchor_box控制了回歸框的大小,我們有時候檢測的是大物體或小物體時,需要調整回歸框的大小的時候,得改一下anchor_box。基於rgb公開的py-faster-rcnn修改an

RFCN python demo.py 檔案 執行後 沒有檢測結果圖片顯示 ?

相關的一個影象檢測演算法,我用的是RFCN,RFCN的時候, python demo_rfcn.py 檔案 執行後 沒有檢測結果圖片顯示,查看了plt.show()函式, 感覺沒有問題啊?因為執行faster rcnn的demo時候,有檢測結果圖,不知道為什麼現在的這

Faster-RCNN-tf使用訓練好的模型驗證測試集 test_net.py

對應原始碼地址:https://github.com/endernewton/tf-faster-rcnn 1、開啟tools目錄下的test_net.py檔案 修改(1): parser.add_argument('--model', dest='model', help='mo

Faster RCNNdemo到訓練自己的資料(1)——前言篇

注意: 一定要保證電腦裝有353.66(CUDA7.5)或369.30(CUDA8.0)以上的N卡驅動,至少保證使用GeForce800系列以上的顯示卡!否則caffe會顯示  “Check failed: error == cudaSuccess (35 vs. 0)”。

faster rcnn 中pascal_voc.py

該部分程式碼功能在於實現了一個pascol _voc的類,該類繼承自imdb,用於負責資料互動部分。 初始化函式 在初始化自身的同時,先呼叫了父類的初始化方法,將imdb _name傳入,例如(‘voc_2007_trainval’),下面是成員變數的初始化: {

Faster RCNNdemo到訓練自己的資料(3)——資料集製作篇

製作資料集可以選擇自己編寫一個demo,也可以直接下載labelImg進行標註。Faster RCNN需要的標註檔案為xml檔案。 我是自己編寫的一個demo,標記出每一張圖片的座標框和分類,生成的文字為txt檔案,也是常見的caffe標註規格,如下圖。 1.將txt

iOS開發項目實戰——Swift實現圖片輪播瀏覽

0.10 上網 timer類 開發項目 cas hub string obj tle 近期開始開發一個新的iOS應用,自己決定使用Swift。進行了幾天之後,發現了一個非常嚴峻的問題。那就是無論是書籍,還是網絡資源,關於Swift的實在是太少了,隨便一

python實現圖片批量剪裁的程序

div imp out code crop path mage .com 程序 1 from PIL import Image 2 import os 3 fin = ‘D:/test‘ 4 fout = ‘D:/test2‘ 5 for file in os.

WWW實現圖片資源顯示以及儲存和本地載入

WWW實現圖片資源顯示以及儲存和本地載入 using UnityEngine; using System.Collections; using System.IO; using UnityEditor; enum GetPicType { DownLoad = 0, Local

java實現圖片的合併縮放

公司有一個需求是 將一個二維碼放在 一個背景圖當中。 因為二維碼是從微信小程式官方獲取的一個byte[],所以需要做一定的調整。   BufferedImage ground = ImageIO.read(this.getClass().getClassLoader().getResource

SSM使用ajax實現圖片上傳刪除功能

圖片上傳與刪除 1.上傳檔案 2.刪除資料,並且刪除對應的檔案 之前寫了一篇部落格記錄了關於修改資料中的圖片上傳 (傳送門),這次用到了直接上傳圖片的功能,對上一篇部落格進行了簡單的優化與改進。 改進問題: 之前刪除該資料時

thinkphp5 結合plupload實現圖片批量上傳

在extend目錄下新增目錄uploader,並新建類Uploads <?php namespace uploader; class Uploads { public static function upfile($file, $path = 'images', $add_domai

.NET使用swfupload實現圖片批量上傳

var swfobject=function(){var X="undefined",P="object",a="visibility:visible",e="visibility:hidden",B="Shockwave Flash",h="ShockwaveFlash.ShockwaveFlash",V=

使用屬性動畫以及自定義view實現圖片的顯示隱藏

效果圖為:點選加號按鈕,加號隱藏,減號旋轉顯示,另外三張圖片也旋轉一定角度顯示 點選減號按鈕,減號隱藏,加號旋轉顯示,另外三張圖片也旋轉一定角度隱藏 首先放五張圖片,使用RelativeLayout佈局,因為可以使這五張圖片重疊起來 customer.xml佈局 &l

關於Django中修改js css檔案但瀏覽器無法及時之改變的問題

今天修改之前實習小夥伴寫的js程式碼的時候,遇到修改後頁面未發生變化的問題。因為我是web開發小白,所以上網查了一波,得以解決~~ 初次進行web工程開發的人可能會碰到這樣的情況:自己在明明對工程上的某個js或css檔案進行了修改,並提交到伺服器上去了

py檔案編譯為pyc(命令指令碼)

http://www.cnblogs.com/dkblog/archive/2009/04/16/1980757.html 其實很簡單, 用 python -m py_compile file.py python -m py_compile /root/src/{f