前提・実現したいこと
自作のデータセットをkerasを使って,google colab環境で転移学習を行い,その学習済み.h5ファイルをローカル環境で実行しようとした際に起きたエラーについてです.
データセットの学習と重みデータの保存までは完了しています.
実現したいことは,ローカル環境で作成した学習済みの.h5ファイルを用いて,読み込みと実行することです.
既存の重みデータでは実行できています.(trained_weights_final.h5は自身で作成した重み)
ローカル環境
Mac OS Python 3.6.9 Keras=2.2.4 tensorflow=1.13.1
発生している問題・エラーメッセージ
ValueError: Cannot create group in read only mode.
該当のソースコード
Python
1 def __init__(self): 2 self.model_path = './model_data/trained_weights_final.h5' #trained_weights_final.h5 3 self.anchors_path = 'model_data/tiny_yolo_anchors.txt' 4 self.classes_path = 'model_data/coco_classes.txt' 5 6 # if args["class"] == 'person': 7 # self.score = 0.6 #0.8 8 # self.iou = 0.6 9 # self.model_image_size = (1024, 576) 10 if args["class"] == 'car': 11 self.score = 0.6 12 self.iou = 0.6 13 self.model_image_size = (512, 288) 14 # if args["class"] == 'bicycle' or args["class"] == 'motorcycle': 15 # self.score = 0.6 16 # self.iou = 0.6 17 # self.model_image_size = (1024, 576) 18 19 self.class_names = self._get_class() 20 self.anchors = self._get_anchors() 21 self.sess = K.get_session() 22 #self.model_image_size = (512, 288) # fixed size or (None, None) small targets:(320,320) mid targets:(960,960) 23 self.is_fixed_size = self.model_image_size != (None, None) 24 self.boxes, self.scores, self.classes = self.generate() 25 26 def _get_class(self): 27 classes_path = os.path.expanduser(self.classes_path) 28 with open(classes_path) as f: 29 class_names = f.readlines() 30 class_names = [c.strip() for c in class_names] 31 #print(class_names) 32 return class_names 33 34 def _get_anchors(self): 35 anchors_path = os.path.expanduser(self.anchors_path) 36 with open(anchors_path) as f: 37 anchors = f.readline() 38 anchors = [float(x) for x in anchors.split(',')] 39 anchors = np.array(anchors).reshape(-1, 2) 40 return anchors 41 42 def generate(self): 43 model_path = os.path.expanduser(self.model_path) 44 assert model_path.endswith('.h5'), 'Keras model must be a .h5 file.' 45 46 self.yolo_model = load_model(model_path, compile=False) 47 print('{} model, anchors, and classes loaded.'.format(model_path)) 48 49 # Generate colors for drawing bounding boxes. 50 hsv_tuples = [(x / len(self.class_names), 1., 1.) 51 for x in range(len(self.class_names))] 52 self.colors = list(map(lambda x: colorsys.hsv_to_rgb(*x), hsv_tuples)) 53 self.colors = list( 54 map(lambda x: (int(x[0] * 255), int(x[1] * 255), int(x[2] * 255)), 55 self.colors)) 56 random.seed(10101) # Fixed seed for consistent colors across runs. 57 random.shuffle(self.colors) # Shuffle colors to decorrelate adjacent classes. 58 random.seed(None) # Reset seed to default. 59 60 # Generate output tensor targets for filtered bounding boxes. 61 self.input_image_shape = K.placeholder(shape=(2, )) 62 boxes, scores, classes = yolo_eval(self.yolo_model.output, self.anchors, 63 len(self.class_names), self.input_image_shape, 64 #score_threshold=self.score, iou_threshold=self.iou 65 ) 66 return boxes, scores, classes 67
試したこと
いろいろ同じエラーを調べていた所,様々な対応策が出てきました.
https://teratail.com/questions/210920
重み保存したファイルの読み込みが必要であること.
https://stackoverflow.com/questions/61926835/error-cannot-create-group-in-read-only-mode
load_weights()を使うこと.
他にも学習済みモデルが壊れていることなども考えられましたが,何回かcolab上で学習は止まりましたが,無事に終わったのでその心配はないと考えています.
様々試しましたがどれも解決に至らなかったため,どなたかご教授お願いします.
個人的には,モデルの読み込み部分に問題があるのでは,と考えています.
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。