Commitc94ac17

committed

edited transfer learning tutorial

1 parent62d02cc commitc94ac17Copy full SHA for c94ac17

File tree

4 files changed

+10

-11

lines changed

machine-learning/image-classifier-using-transfer-learning

4 files changed

+10

-11

lines changed

`‎machine-learning/image-classifier-using-transfer-learning/results/MobileNetV2_finetune_last5-loss-0.63.h5`

23.8 MB

Binary file not shown.

`‎machine-learning/image-classifier-using-transfer-learning/results/MobileNetV2_finetune_last5_less_lr-loss-0.45-acc-0.86.h5`

-12.3 MB

Binary file not shown.

`‎machine-learning/image-classifier-using-transfer-learning/test.py`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -5,7 +5,7 @@`
`5`	`5`	`# constructs the model`
`6`	`6`	`model=create_model(input_shape=IMAGE_SHAPE)`
`7`	`7`	`# load the optimal weights`
`8`		`-model.load_weights("results/MobileNetV2_finetune_last5_less_lr-loss-0.45-acc-0.86.h5")`
	`8`	`+model.load_weights("results/MobileNetV2_finetune_last5-loss-0.66.h5")`
`9`	`9`
`10`	`10`	`validation_steps_per_epoch=np.ceil(validation_generator.samples/batch_size)`
`11`	`11`	`# print the validation loss & accuracy`

`‎machine-learning/image-classifier-using-transfer-learning/train.py`

Lines changed: 9 additions & 10 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,11 +1,10 @@`
`1`		`-`
`2`	`1`	`importtensorflowastf`
`3`		`-fromkeras.modelsimportModel`
`4`		`-fromkeras.applicationsimportMobileNetV2,ResNet50,InceptionV3# try to use them and see which is better`
`5`		`-fromkeras.layersimportDense`
`6`		`-fromkeras.callbacksimportModelCheckpoint,TensorBoard`
`7`		`-fromkeras.utilsimportget_file`
`8`		`-fromkeras.preprocessing.imageimportImageDataGenerator`
	`2`	`+fromtensorflow.keras.modelsimportModel`
	`3`	`+fromtensorflow.keras.applicationsimportMobileNetV2,ResNet50,InceptionV3# try to use them and see which is better`
	`4`	`+fromtensorflow.keras.layersimportDense`
	`5`	`+fromtensorflow.keras.callbacksimportModelCheckpoint,TensorBoard`
	`6`	`+fromtensorflow.keras.utilsimportget_file`
	`7`	`+fromtensorflow.keras.preprocessing.imageimportImageDataGenerator`
`9`	`8`	`importos`
`10`	`9`	`importpathlib`
`11`	`10`	`importnumpyasnp`
`@@ -65,7 +64,7 @@ def create_model(input_shape):`
`65`	`64`	`# print the summary of the model architecture`
`66`	`65`	`model.summary()`
`67`	`66`
`68`		`-# training the model usingrmsprop optimizer`
	`67`	`+# training the model usingadam optimizer`
`69`	`68`	`model.compile(loss="categorical_crossentropy",optimizer="adam",metrics=["accuracy"])`
`70`	`69`	`returnmodel`
`71`	`70`
`@@ -81,8 +80,8 @@ def create_model(input_shape):`
`81`	`80`	`model_name="MobileNetV2_finetune_last5"`
`82`	`81`
`83`	`82`	`# some nice callbacks`
`84`		`-tensorboard=TensorBoard(log_dir=f"logs/{model_name}")`
`85`		`-checkpoint=ModelCheckpoint(f"results/{model_name}"+"-loss-{val_loss:.2f}-acc-{val_acc:.2f}.h5",`
	`83`	`+tensorboard=TensorBoard(log_dir=os.path.join("logs",model_name))`
	`84`	`+checkpoint=ModelCheckpoint(os.path.join("results",f"{model_name}"+"-loss-{val_loss:.2f}.h5"),`
`86`	`85`	`save_best_only=True,`
`87`	`86`	`verbose=1)`
`88`	`87`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commitc94ac17

File tree

4 files changed

4 files changed

`‎machine-learning/image-classifier-using-transfer-learning/results/MobileNetV2_finetune_last5-loss-0.63.h5`

`‎machine-learning/image-classifier-using-transfer-learning/results/MobileNetV2_finetune_last5_less_lr-loss-0.45-acc-0.86.h5`

`‎machine-learning/image-classifier-using-transfer-learning/test.py`

`‎machine-learning/image-classifier-using-transfer-learning/train.py`

0 commit comments