์ด ๋ช
๋ น์ ์ฌ์ฉํ์ฌ ์ฃผ์์ผ๋ก ๋ด ๋ฐ์ดํฐ์ธํธ๋ฅผ ํ๋ จํ๋ ค๊ณ ํฉ๋๋ค.
python flow --model cfg/tiny-yolo-voc-4c.cfg --load bin/tiny-yolo-voc.weights --train --annotation custom/annotations --dataset custom/all --gpu 0.8 --epoch 500
๋ด ์ฃผ์ ํด๋(๋ชจ๋ xml)
๋ด ๋ฐ์ดํฐ ์ธํธ ํด๋(๋ชจ๋ jpg)
์ค๋ฅ:
Statistics:
Dataset size: 0
Dataset of 0 instance(s)
Traceback (most recent call last):
File "flow", line 6, in <module>
cliHandler(sys.argv)
File "C:\Users\msi\Desktop\VSCode Folder\AI Stuff\YOLO Darkflow\darkflow-gpu\darkflow\cli.py", line 33, in cliHandler
print('Enter training ...'); tfnet.train()
File "C:\Users\msi\Desktop\VSCode Folder\AI Stuff\YOLO Darkflow\darkflow-gpu\darkflow\net\flow.py", line 39, in train
for i, (x_batch, datum) in enumerate(batches):
File "C:\Users\msi\Desktop\VSCode Folder\AI Stuff\YOLO Darkflow\darkflow-gpu\darkflow\net\yolo\data.py", line 102, in shuffle
batch_per_epoch = int(size / batch)
ZeroDivisionError: division by zero
๊ฐ์ ์ค๋ฅ๊ฐ ๋ฐ์ํฉ๋๋ค. VOC ๋ฐ์ดํฐ ์ธํธ๋ฅผ ์ฌ์ฉํ์ฌ ์ฌํ๋ จ์ ์๋ํ์ต๋๋ค. ํด๊ฒฐํ์ จ๋์?
Statistics:
Dataset size: 0
Dataset of 0 instance(s)
Traceback (most recent call last):
File "/usr/local/bin/flow", line 6, in <module>
cliHandler(sys.argv)
File "/videoEval/darkflow/darkflow/cli.py", line 29, in cliHandler
print('Enter training ...'); tfnet.train()
File "/videoEval/darkflow/darkflow/net/flow.py", line 39, in train
for i, (x_batch, datum) in enumerate(batches):
File "/videoEval/darkflow/darkflow/net/yolo/data.py", line 102, in shuffle
batch_per_epoch = int(size / batch)
ZeroDivisionError: division by zero
์์ @humanova์ "darkflow / darkflow / ๊ทธ๋ฌผ / yolov2 / data.py"์ด๋ฏธ์ง์ ๊ฒฝ๋ก๋ฅผ ์ธ์ ํด๋ณด์ญ์์ค, ๋น์ ์ ์ค์ ๋ฌด์์ธ์ง๊ฐ .. ์๊ฒ๋ฉ๋๋ค! ๋ค์ ์ฝ๋๋ฅผ ์ฐธ์กฐํ์ญ์์ค.
jpg = chunk[0]; w, h, allobj_ = chunk[1]
allobj = deepcopy(allobj_)
path = os.path.join(self.FLAGS.dataset, jpg)
**print("Processed : {}".format(path))**
img = self.preprocess(path, allobj)
@naren142 ๋ค, ์ฒ๋ฆฌ๋ ์ด๋ฏธ์ง๋ฅผ ์ถ๋ ฅํ๋ฉด์ ํธ๋ ์ด๋ํ๋ฉด์ ์ด ๋ฌธ์ ๋ฅผ ๋ง๋ ๋ ๋ฐ๋ก ์์ ํ์ต๋๋ค.
๋ํ darkflow/utils/pascal_voc_clean_xml.py์์ ์ด ์ฝ๋ ์ค์ ๋ณ๊ฒฝ
annotations = glob.glob('*.xml')
๋ฐ์ดํฐ๊ฐ ์๋ ์ฃผ์์ผ๋ก ์ธํด ๋ฐ์ํ ์ค๋ฅ์ ๋๋ค. (์ด๋ฏธ์ง์ ๋ํ ์ฃผ์์ด ์์ง๋ง ์ด๋ฏธ์ง๊ฐ ์ฌ๋ผ์ง) ํด๋น annotation.xml์ ์ญ์ ํ ๋ค์ ์๋ํ์ต๋๋ค! ๊ฐ์ฌ ํด์!
๊ฐ์ ๋ฌธ์ ๊ฐ์์์ต๋๋ค. darkflow์๋ txt ์ฃผ์์ด ์๋ xml ํ์ผ์ด ํ์ํ๋ค๋ ๊ฒ์ ์์๋์ต๋๋ค.
๊ฐ์ฅ ์ ์ฉํ ๋๊ธ
@naren142 ๋ค, ์ฒ๋ฆฌ๋ ์ด๋ฏธ์ง๋ฅผ ์ถ๋ ฅํ๋ฉด์ ํธ๋ ์ด๋ํ๋ฉด์ ์ด ๋ฌธ์ ๋ฅผ ๋ง๋ ๋ ๋ฐ๋ก ์์ ํ์ต๋๋ค.
๋ํ darkflow/utils/pascal_voc_clean_xml.py์์ ์ด ์ฝ๋ ์ค์ ๋ณ๊ฒฝ
annotations = glob.glob('*.xml')
๋ฐ์ดํฐ๊ฐ ์๋ ์ฃผ์์ผ๋ก ์ธํด ๋ฐ์ํ ์ค๋ฅ์ ๋๋ค. (์ด๋ฏธ์ง์ ๋ํ ์ฃผ์์ด ์์ง๋ง ์ด๋ฏธ์ง๊ฐ ์ฌ๋ผ์ง) ํด๋น annotation.xml์ ์ญ์ ํ ๋ค์ ์๋ํ์ต๋๋ค! ๊ฐ์ฌ ํด์!