이전에꺼는 save를 못해먹겠어서 (checkpoint를 pb로 변환하거나 해야 하는데 그것도 안되고...)
다른 것 찾는 중
[링크 : https://www.kaggle.com/code/suraj520/mobilenet-v2-ssd-scratch-without-tfod]
위에껀 아래와 같이 sequential 이라던가 이런게 없어서.. 또 저장안되는거 아닌가 걱정중..
def create_model(): model = tf.keras.Sequential([ keras.layers.Dense(512, activation='relu', input_shape=(784,)), keras.layers.Dropout(0.2), keras.layers.Dense(10) ]) model.compile(optimizer='adam', loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True), metrics=[tf.keras.metrics.SparseCategoricalAccuracy()]) return model # Create a basic model instance model = create_model() |
[링크 : https://www.tensorflow.org/tutorials/keras/save_and_load?hl=ko]
[링크 : https://www.tensorflow.org/hub/exporting_tf2_saved_model?hl=ko]
'프로그램 사용 > yolo_tensorflow' 카테고리의 다른 글
VGG-16 / VGG-19 (0) | 2024.01.11 |
---|---|
MobileNetV2 SSD FPN-Lite (0) | 2024.01.11 |
ssd-mobilenetv2 on jupyter notebook (2) | 2024.01.10 |
텐서플로우 v1 을 v2로 마이그레이션은 실패 -_- (0) | 2024.01.10 |
골빈해커의 3분 딥러닝 github (0) | 2024.01.10 |