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