본문 바로가기

STUDY/인공지능 & 딥러닝

[keras] GPU 메모리 사용량 제한하기 https://groups.google.com/forum/#!topic/keras-users/MFUEY9P1sc8 위 링크에서 찾을 수 있었는데 정리하자면 아래의 소스를 처음 부분에 입력해주면 GPU 메모리 전체를 사용하지 않는다. gpu_fraction 의 값에 따라서 전체 메모리 중 몇 퍼센트를 이용할 것인지 설정할 수 있다. Python 3.5, Keras 2.1.2 버전에서 실행하였다. import os import tensorflow as tf import keras.backend.tensorflow_backend as KK def get_session(gpu_fraction=0.3): '''Assume that you have 6GB of GPU memory and want to allocat.. 더보기
[딥러닝] Deep Learning을 이용한 객체 검출 알고리즘 소개 출처 : http://tech-blog.abeja.asia/entry/object-detection-summary 一般物体検出アルゴリズムの紹介 (일반물체검출 알고리즘의 소개)今回CNNを用いた一般物体検出アルゴリズムの有名な論文を順を追って説明します。コンピュータビジョンの分野において、一般物体検出とは下記の図のように、ある画像の中から定められた物体の位置とカテゴリー(クラス)を検出することを指します。[6]より引用Deep Learningアルゴリズムの発展によって、一般物体認識の精度は目まぐるしい勢いで進歩しております。 そこで今回はDeep Learning(CNN)を応用した、一般物体検出アルゴリズムの有名な論文を説明したいと思います。R-CNN (Regions with CNN features) (CVPR 2014) [1]かの有名なCNNの論文[8.. 더보기