1. 程式人生 > >ChainerCV下用自己的資料集訓練Faster RCNN

ChainerCV下用自己的資料集訓練Faster RCNN

一、ChainerCV

        ChainerCV,一個用於計算機視覺的深度學習實用庫。這個庫旨在通過 Chainer 簡易化計算機視覺的訓練和深度學習模型應用的過程。它包含計算機視覺模型的高質量實現,以及開展計算機視覺研究的必備工具集。當前,ChainerCV 提供了目標檢測和語義分割模型(Faster R-CNN、SSD 和 SegNet)的實現。

        https://github.com/chainer/chainercv

二、修改原始程式碼

        1,chainercv/datasets/voc/voc_utils.py中

(1)更改voc_bbox_label_names,改為自己資料庫的類標;增加oc_semantic_segmentation_label_colors,和類標相對應

        2,chainercv/datasets/voc/voc_bbox_dataset.py中,

(1)第8行from chainercv.datasets.voc import voc_utils 改為 import voc_utils;

(2)在get_example中,將obj.find('name').text.lower().strip()改為obj.find('name').text.strip(),因為.lower()為把所有的namee變為小寫,而我們的資料集的label name是區分大小寫的

        3,examples/faster_rcnn/train.py中,

(1)檔案開頭加入 import sys,sys.path.append(r'/home/wang/Development/chainercv/chainercv/datasets/voc');並將

       from chainercv.datasets import voc_bbox_label_names

       from chainercv.datasets import VOCBboxDataset

改為from voc_utils import voc_bbox_label_names

       from voc_bbox_dataset import VOCBboxDataset

這樣匯入的label name為我們修改後的,而不是原始VOC的;

(2)並在main()中

train_data = VOCBboxDataset(split='trainval', year='2007')加入路徑變為

train_data = VOCBboxDataset('/home/wang/Development/VOCdevkit2007/VOC2007',split='trainval', year='2007'),

同樣的test_data = VOCBboxDataset(data_dir='/home/wang/Development/VOCdevkit2007/VOC2007',split='test', year='2007',use_difficult=True, return_difficult=True);

這裡最重要,加了自己資料庫的路徑後,就直接匯入自己的資料庫,若沒有路徑,會在VOC連結下載VOC資料庫,就會導致錯誤。

         4,examples/faster_rcnn/demo.py中,

(1) 檔案開頭加入 import sys,sys.path.append(r'/home/wang/Development/chainercv/chainercv/datasets/voc');

(2) 並將 from chainercv.datasets import voc_bbox_label_names 改為 from voc_utils import voc_bbox_label_names。和train的原因一樣

三,訓練 

python train.py --iteration 70000  --gpu 0

四,測試

python demo.py --gpu 1 --pretrained_model /home/wang/Development/chainercv/examples/faster_rcnn/result/snapshot_model.npz /home/wang/Development/VOCdevkit2007/VOC2007/JPEGImages/000102.jpg

注:在換了電腦後訓練和測試出現問題

這是因為版本不相容導致GPU出問題,用命令sudo pip install cupy==2.0.0將cupy降為2.0.0,問題解決。此時版本資訊為:


相關推薦

ChainerCV自己資料訓練Faster RCNN

一、ChainerCV         ChainerCV,一個用於計算機視覺的深度學習實用庫。這個庫旨在通過 Chainer 簡易化計算機視覺的訓練和深度學習模型應用的過程。它包含計算機視覺模型的高質量實現,以及開展計算機視覺研究的必備工具集。當前,ChainerCV

自己資料訓練faster-rcnn時出現的一些問題及總結(五)

關於faster-rcnn的原理訓練流程及其資料集的製備,已經都已完成。那麼問題來了,“自己製備的資料集,比如說自己為了增強資料集,進行圖片翻轉操作,然後記錄座標這樣的資料集,在訓練的過程中出現的問題”一、在呼叫append_flipped_images函式時出現: asse

windows使用自己製作的資料訓練faster-rcnn(tensorflow版)用於目標檢測

步驟一 步驟二 步驟三 用你的Annotations,ImagesSets和JPEGImages替換…\Faster-RCNN-TensorFlow-Python3.5\data\VOCdevkit2007\VOC2007中對應資料夾

使用自己資料訓練faster-rcnn

在安裝完成caffe後,並且編譯完成github上的faster-rcnn python版之後,可以採用自己的資料來訓練faster-rcnn了。 一,檔案修改: 1,在py-faster-rcnn目錄下,找到lib/datasets/pascal_vo

ImageNet的資料訓練Faster R-CNN

轉自:http://blog.csdn.net/jiajunlee/article/details/50470897 訓練結果 第一次訓練,ZF,30W張圖片,200類,迭代80k40K,mAP: 18.0% 問題彙總 總結: ImageNet訓練資料集部分標註缺失

自己資料訓練Mask-RCNN實現過程中的坑

本文僅僅是自己實現過程的筆記記錄,僅僅用來交流的。 在網上大量蒐集資料後,實現Mask-RCNN,但是過程中還是出現了很多很多的問題,所以將過程記錄如下,方便日後學習。 一、實驗前準備 1. COCO資料集 COCO的 全稱是Common Objects in COn

使用coco資料faster rcnn類方法訓練出錯解決

問題:在caffe框架下,使用coco資料集進行faster rcnn類方法訓練,得到如下錯誤:  File "/data/zn/light_head_rcnn/script/py-RFCN-priv/tools/../lib/rpn/anchor_target_layer.

製作VOC2007格式資料用於Faster-RCNN訓練

轉自http://blog.csdn.net/sinat_30071459/article/details/50723212,自己加入了自己製作資料集時遇到的一些問題,和想法 首先錄一段想製作成資料集的目標的視訊,截成一幀一幀的圖片,作為資料集的原材料。 0.資料夾

Faster-RCNN+ZF自己資料訓練模型(Python版本and MATLAB版本)

說明:本博文假設你已經做好了自己的資料集,該資料集格式和VOC2007相同。下面是訓練前的一些修改。 Faster-RCNN原始碼下載地址: 本文用到的是版本,在Linux下執行。 準備工作: 1.配置caffe   

Faster-RCNN+ZF自己資料訓練模型(Python版本)

說明:本博文假設你已經做好了自己的資料集,該資料集格式和VOC2007相同。下面是訓練前的一些修改。 Faster-RCNN原始碼下載地址: 本文用到的是Python版本,在Linux下執行。 準備工作: 1.配置caffe      這個不多說,網上教程很多。

使用pytorch版faster-rcnn訓練自己資料

使用pytorch版faster-rcnn訓練自己資料集 引言 faster-rcnn pytorch程式碼下載 訓練自己資料集 接下來工作 參考文獻 引言 最近在復現目標檢測程式碼(師兄強烈推薦F

自己資料訓練faster-rcnn---cpu

模型下載: 修改模型中的部分程式碼: 因為是用CPU跑模型,所以要對程式碼進行適當修改。 用自己的資料集訓練faster-rcnn: 1.擁有訓練的資料以及資料對應的xml檔案。 我這裡用的是以前人臉座標資料,用的標註工具是LableImg,當然還有其他的標

tensorflow 自己資料訓練CNN模型

最近用tensorflow訓練自己的模型的時候發現,tensorflow官網上所給的例子,都是用處理好資料格式的mnist資料或者其他格式的資料,所以在訓練自己的模型的時候的第一步就卡住了。所以上網搜尋了相關的資料之後便得出了相關的解決方案(有好幾種,這裡只說明一種,另外

windows10+GPUcaffe資料Lmdb格式製作+訓練自己資料

最近做人臉識別專案,想用到caffe訓練自己的資料,電腦作業系統為win10+GPU,這裡對caffe-windows配置、資料集製作、訓練資料都做一些介紹。(無GPU配置的看我這個部落格)。如果你用的是vs2015,那麼下面介紹的caffe不適合,要用BVLC wind

自己資料訓練tf-ssd模型

資料集製作因為老闆接的豐田的一個專案,工廠那邊要求能識別出雨天打傘的行人、交通錐形桶、躺在地上的人等,PASCAL VOC的資料集類別裡沒這些,是滿足不了他們要求了,所以要去製作資料集訓練網路。我們去現場拍了些視訊,然後用我之前寫的一個指令碼解析,得到原始圖片,拿給實驗室的學

Faster-RCNN 訓練自己資料的坑記錄

主要照這篇部落格進行訓練配置,因為沒有GPU所以好多坑,CPU訓練可以參見這篇部落格 正所謂,跑通了的都一樣,錯誤千萬樣。按照教程來也是坑多 訓練: python train_faster_rcnn_alt_opt.py --net_name ZF --weights

在detectron自己資料訓練Faster-Rcnn+FPN

我的系統ubuntu16.04程式碼放在github上https://github.com/withyou1771/Detectron_FocalLoss歡迎大家點亮星星~~~安裝caffe21、Required Dependenciessudo apt-get update

py-faster-rcnn訓練自己資料需要修改的引數

faster rcnn預設有三種網路模型ZF(小)、VGG_CNN_M_1024(中)、VGG16 (大) 訓練圖片大小為500*500,類別數1。 修改VGG_CNN_M_1024模型配置

TensorFlow實戰:Chapter-9(DeepLabv3+在自己資料訓練)

基本配置 資料集處理 我的資料集是3分類問題,但因為資料集的保密協議,在後面的demo中我沒有放出原圖片,我會盡量將訓練細節寫出來。為了方便記錄,我又使用了CamVid資料集(從這裡下載)測試了一下。 資料集處理分成三大步: 標註資料 製作

自己資料訓練Faster-RCNN,tensorflow版本(一)

我用的Faster-RCNN是tensorflow版本,fork自githubFaster-RCNN_TF 1.1、環境配置 按照該專案中的README.md ,將需要的幾個依賴cython, python-opencv, easydict都安裝