im2col
์ด๋ฏธ์ง๋ฅผ columns์ผ๋ก ๋ณํํด์ฃผ๋ ๊ฒ์ ๋งํฉ๋๋ค.
im2col_get_pixel
float im2col_get_pixel(float *im, int height, int width, int channels,
int row, int col, int channel, int pad)
{
row -= pad;
col -= pad;
if (row < 0 || col < 0 ||
row >= height || col >= width) return 0;
return im[col + width*(row + height*channel)];
}
ํจ์ ์ด๋ฆ: im2col_get_pixel
์ ๋ ฅ:
im: ์ด๋ฏธ์ง ๋ฐ์ดํฐ๋ฅผ ๊ฐ๋ฆฌํค๋ ํฌ์ธํฐ(float *)
height: ์ด๋ฏธ์ง ๋์ด(int)
width: ์ด๋ฏธ์ง ๋๋น(int)
channels: ์ด๋ฏธ์ง ์ฑ๋ ์(int)
row: ํฝ์ ์ ์ธ๋ก ์์น(int)
col: ํฝ์ ์ ๊ฐ๋ก ์์น(int)
channel: ํฝ์ ์ด ์์นํ ์ฑ๋(int)
pad: ํจ๋ฉ ํฌ๊ธฐ(int)
๋์:
์ ๋ ฅ์ผ๋ก ์ฃผ์ด์ง ์์น(row, col, channel)์ ํด๋นํ๋ ํฝ์ ๊ฐ์ ๋ฐํํ๋ค.
๋จ, ํจ๋ฉ(pad)์ด ์ ์ฉ๋์ด ์๋ ๊ฒฝ์ฐ, ํจ๋ฉ ํฌ๊ธฐ๋งํผ ์์น๋ฅผ ์กฐ์ ํ์ฌ ์ด๋ฏธ์ง ๋ฐ์ดํฐ๋ฅผ ๊ฐ์ ธ์จ๋ค.
๋ง์ฝ ๊ฐ์ ธ์ฌ ๋ฐ์ดํฐ๊ฐ ์ด๋ฏธ์ง์ ๋ฒ์๋ฅผ ๋ฒ์ด๋๋ ๊ฒฝ์ฐ, 0์ ๋ฐํํ๋ค.
์ค๋ช :
์ด๋ฏธ์ง ๋ฐ์ดํฐ๋ฅผ 2์ฐจ์ ๋ฐฐ์ด ํํ๋ก ๋ณํํ๋ im2col ์ฐ์ฐ์์, ์ฃผ์ด์ง ์์น์ ๋์ํ๋ ๋ฐ์ดํฐ๋ฅผ ๊ฐ์ ธ์ค๊ธฐ ์ํด ์ฌ์ฉ๋๋ค.
์ด ํจ์๋ ํด๋น ์์น(row, col, channel)์์์ ๊ฐ์ ๊ฐ์ ธ์ค๋ ์ญํ ์ ํ๋ค.
์ด๋ฏธ์ง ๋ฐ์ดํฐ๋ 1์ฐจ์ ๋ฐฐ์ด๋ก ์ ์ฅ๋์ด ์๊ธฐ ๋๋ฌธ์, row, col, channel ์ ๋ณด๋ฅผ ์ด์ฉํ์ฌ ์ธ๋ฑ์ค๋ฅผ ๊ณ์ฐํ์ฌ ๊ฐ์ ๊ฐ์ ธ์จ๋ค.
๋จ, ํจ๋ฉ์ด ์ ์ฉ๋ ๊ฒฝ์ฐ, row์ col์์ ํจ๋ฉ์ ํฌ๊ธฐ(pad)๋ฅผ ๋บ ๊ฐ์ ์ฌ์ฉํ์ฌ ์ธ๋ฑ์ค๋ฅผ ๊ณ์ฐํ๋ค.
im2col_cpu
//From Berkeley Vision's Caffe!
//https://github.com/BVLC/caffe/blob/master/LICENSE
void im2col_cpu(float* data_im,
int channels, int height, int width,
int ksize, int stride, int pad, float* data_col)
{
int c,h,w;
int height_col = (height + 2*pad - ksize) / stride + 1;
int width_col = (width + 2*pad - ksize) / stride + 1;
int channels_col = channels * ksize * ksize;
for (c = 0; c < channels_col; ++c) {
int w_offset = c % ksize;
int h_offset = (c / ksize) % ksize;
int c_im = c / ksize / ksize;
for (h = 0; h < height_col; ++h) {
for (w = 0; w < width_col; ++w) {
int im_row = h_offset + h * stride;
int im_col = w_offset + w * stride;
int col_index = (c * height_col + h) * width_col + w;
data_col[col_index] = im2col_get_pixel(data_im, height, width, channels,
im_row, im_col, c_im, pad);
}
}
}
}
ํจ์ ์ด๋ฆ: im2col_cpu
์ ๋ ฅ:
data_im: float ํฌ์ธํฐ, ์ ๋ ฅ ์ด๋ฏธ์ง ๋ฐ์ดํฐ ํฌ์ธํฐ
channels: int, ์ ๋ ฅ ์ด๋ฏธ์ง ์ฑ๋ ์
height: int, ์ ๋ ฅ ์ด๋ฏธ์ง ๋์ด
width: int, ์ ๋ ฅ ์ด๋ฏธ์ง ๋๋น
ksize: int, ํํฐ(์ปค๋) ํฌ๊ธฐ
stride: int, ์คํธ๋ผ์ด๋ ํฌ๊ธฐ
pad: int, ํจ๋ฉ ํฌ๊ธฐ
data_col: float ํฌ์ธํฐ, ๋ณํ๋ ์ด๋ฏธ์ง ๋ฐ์ดํฐ ํฌ์ธํฐ
๋์:
์ ๋ ฅ ์ด๋ฏธ์ง๋ฅผ ํํฐ ํฌ๊ธฐ, ์คํธ๋ผ์ด๋, ํจ๋ฉ ์ ๋ณด๋ฅผ ๊ธฐ๋ฐ์ผ๋ก im2col ๋ฐฉ์์ผ๋ก ๋ณํํ์ฌ data_col์ ์ ์ฅํ๋ ํจ์์ ๋๋ค.
Caffe๋ผ๋ ๋จธ์ ๋ฌ๋ ํ๋ ์์ํฌ์์ ๊ฐ์ ธ์จ ์ฝ๋๋ก, ์ ๋ ฅ ์ด๋ฏธ์ง์ ๊ฐ ์ฑ๋์์ ํํฐ์ ํฌ๊ธฐ๋งํผ ์์ง์ฌ๊ฐ๋ฉฐ ์ฌ๋ผ์ด๋ฉ ์๋์ฐ๋ฅผ ๋ง๋ค๊ณ , ์๋์ฐ ๋ด์ ๊ฐ์ ์ผ๋ ฌ๋ก ๋์ด๋์ ๋ค์, data_col์ ์ ์ฅํฉ๋๋ค.
์ค๋ช :
height_col: int, im2col ๋ณํ ํ ์ถ๋ ฅ ์ด๋ฏธ์ง ๋์ด
width_col: int, im2col ๋ณํ ํ ์ถ๋ ฅ ์ด๋ฏธ์ง ๋๋น
channels_col: int, im2col ๋ณํ ํ ์ถ๋ ฅ ์ด๋ฏธ์ง ์ฑ๋ ์
c: int, channels_col ๋ด ํ์ฌ ์ฑ๋ ์ธ๋ฑ์ค
w_offset: int, ํ์ฌ ์ฑ๋์์ ํํฐ์ ๋๋น ์คํ์
h_offset: int, ํ์ฌ ์ฑ๋์์ ํํฐ์ ๋์ด ์คํ์
c_im: int, ํ์ฌ ์ฑ๋์์ ์ ๋ ฅ ์ด๋ฏธ์ง ์ฑ๋ ์ธ๋ฑ์ค
h: int, ์ถ๋ ฅ ์ด๋ฏธ์ง์ ํ์ฌ ๋์ด ์ธ๋ฑ์ค
w: int, ์ถ๋ ฅ ์ด๋ฏธ์ง์ ํ์ฌ ๋๋น ์ธ๋ฑ์ค
im_row: int, ์ ๋ ฅ ์ด๋ฏธ์ง ๋ด ํ์ฌ ์์น์ ๋์ด ์ธ๋ฑ์ค
im_col: int, ์ ๋ ฅ ์ด๋ฏธ์ง ๋ด ํ์ฌ ์์น์ ๋๋น ์ธ๋ฑ์ค
col_index: int, ์ถ๋ ฅ ์ด๋ฏธ์ง ๋ด ํ์ฌ ์์น์ ์ธ๋ฑ์ค
im2col_get_pixel: im2col ๋ณํ์ ํ์ฌ ์์น์์์ ํฝ์ ๊ฐ์ ๊ฐ์ ธ์ค๋ ํจ์
Last updated
Was this helpful?