NuclearDataset train๊ณผ์ ์์ ์ค๋ฅ๊ฐ ๋ฐ์ํฉ๋๋ค.
์ ๋ balloon_model.train(...) ์คํ์, ๊ฐ์ ๋ฌธ์ ๊ฐ ๋ฐ์ํ์ฌ ๋ค์๊ณผ ๊ฐ์ด ์์ ํ์ฌ ์ฌ์ฉํ๊ณ ์์ต๋๋ค.1) ๋ค์ด๋ก๋ ๋ฐ์ Mask_RCNN/mrcnn/model.py ์์ use_multiprocessing=False ๋ก ์์ ํ์ฌ ์ ์ฅ2) python setup.py install ๋ช
๋ น์ ์ฌ์ฉํ์ฌ Mask_RCNN ์ฌ๋ฐฐํฌํน์, ํด๋น ๊ธฐ๋ฅ์ด ์ฑ๋ฅ ์ ๋ฌธ์ ๊ฐ ๋๋ค๋ฉด, ๋์ค์ ๋ค์ use_multiprocessing=True๋ก ์์ ํ์ฌ ์ฌ๋ฐฐํฌํ๋ฉด ๋ ๊ฒ ๊ฐ๋ค์. ์ ์๋์ ํ์ธํ์ผ๋ฉฐ, train() ์์์๊ฐ์ 12๋ถ 58์ด ๊ฑธ๋ ธ๋ค์. (use_multiprocessing=True ๋ก ํ์ ๊ฒฝ์ฐ ์๊ฐ์ด ์ผ๋ง๋ ๊ฑธ๋ฆด๊น์?)(GPU๋ Titan X(๋ฉ๋ชจ๋ฆฌ 12GB) ์ฌ์ฉ ์ค์
๋๋ค.) #---[๊ด๋ จ ์์ค์ฝ๋]-------- print("Training network heads")balloon_model.train(dataset_train, dataset_val, learning_rate=train_config.LEARNING_RATE, epochs=30, layers='heads') #---[์๋ฌ๋ด์ฉ]------------- #~/anaconda3/envs/tf113/lib/python3.6/site-packages/mask_rcnn-2.1-py3.6.egg/mrcnn/model.py in train(self, train_dataset, val_dataset, learning_rate, epochs, layers, augmentation, custom_callbacks, no_augmentation_sources)# 2372 max_queue_size=100,# 2373 workers=workers,#-> 2374 use_multiprocessing=True,#:# EOFError#Process Process-15:# File "../anaconda3/envs/tf113/lib/python3.6/multiprocessing/connection.py", line 407, in _recv_bytes# buf = self._recv(4)#EOFError #-----[์์ ํ ์ ์ ๋์ํจ: ์์์๊ฐ 12๋ถ 58์ด]----------- : Epoch 30/30 61/61 [==============================] - 22s 361ms/step - loss: 0.1355 - rpn_class_loss: 0.0026 - rpn_bbox_loss: 0.0274 - mrcnn_class_loss: 0.0214 - mrcnn_bbox_loss: 0.0211 - mrcnn_mask_loss: 0.0630 - val_loss: 0.6885 - val_rpn_class_loss: 0.0778 - val_rpn_bbox_loss: 0.3417 - val_mrcnn_class_loss: 0.0657 - val_mrcnn_bbox_loss: 0.0819 - val_mrcnn_mask_loss: 0.1214 time: 12min 58s