Page 11 -
P. 11
1.9 15줄의 짧은 프로그램으로 이미지를
DEEP LEARNING
학습시키기
일단 실제로 학습하는 프로그램을 작성해봅시다. 다음과 같은 프로그램으로 Deel에서 이미지 데
이터셋을 학습시킬 수 있습니다.
코드 1-7 tiny4.py
1: from deel import *
2: from deel.network import *
3: from deel.commands import *
4:
5: deel = Deel()
6: nin = NetworkInNetwork()
7:
8: InputBatch(train="data/train.txt",
9: val="data/test.txt")
10:
11: def workout(x,t):
12: nin.classify(x)
13: return nin.backprop(t)
14:
15: BatchTrain(workout)
이때 train.txt나 text.txt에는 학습 데이터셋과 테스트 데이터셋이 들어 있습니다.
코드 1-8 train.txt의 내용(예)
/Users/shi3z/deel/data/images/image0000000.jpg 0
/Users/shi3z/deel/data/images/image0000001.jpg 0
/Users/shi3z/deel/data/images/image0000002.jpg 0
/Users/shi3z/deel/data/images/image0000003.jpg 0
/Users/shi3z/deel/data/images/image0000004.jpg 0
/Users/shi3z/deel/data/images/image0000005.jpg 0
.
.
.
/Users/shi3z/deel/data/images/image0009120.jpg 101
/Users/shi3z/deel/data/images/image0009121.jpg 101
/Users/shi3z/deel/data/images/image0009122.jpg 101
042
deeplearing_07.indd 42 2018-04-11 오전 8:12:10