spacecruft
/
satnogs-wut
0
0
Fork 0
Code Issues 6 Releases Activity

train GMSK

master
ml server 2020-01-29 20:15:41 -07:00
parent 7eb8676f11
commit d96585e514
1 changed files with 10 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
notebooks/wut-train.ipynb
View File

@ -149,7 +149,7 @@
"metadata": {},
"outputs": [],
"source": [
"ENCODING='PSK31'"
"ENCODING='GMSK'"
]
},
{
@ -221,27 +221,11 @@
"print('total training good images:', num_train_good)\n",
"print('total training bad images:', num_train_bad)\n",
"print(\"--\")\n",
"print(\"Total training images:\", total_train)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"print(\"Total training images:\", total_train)\n",
"print('total validation good images:', num_val_good)\n",
"print('total validation bad images:', num_val_bad)\n",
"print(\"--\")\n",
"print(\"Total validation images:\", total_val)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"print(\"Total validation images:\", total_val)\n",
"print(\"Reduce training and validation set when testing\")\n",
"#total_train = 100\n",
"#total_val = 100\n",
@ -257,7 +241,7 @@
"metadata": {},
"outputs": [],
"source": [
"batch_size = 64\n",
"batch_size = 32\n",
"epochs = 16"
]
},
@ -267,12 +251,8 @@
"metadata": {},
"outputs": [],
"source": [
"# Half size\n",
"IMG_HEIGHT = 804\n",
"IMG_WIDTH= 416\n",
"# Full size, machine barfs probably needs more RAM\n",
"#IMG_HEIGHT = 832\n",
"#IMG_WIDTH = 1606"
"IMG_WIDTH = 832\n",
"IMG_HEIGHT = 1606"
]
},
{
@ -281,9 +261,7 @@
"metadata": {},
"outputs": [],
"source": [
"train_image_generator = ImageDataGenerator(\n",
" rescale=1./255\n",
")"
"train_image_generator = ImageDataGenerator( rescale=1./255 )"
]
},
{
@ -292,9 +270,7 @@
"metadata": {},
"outputs": [],
"source": [
"val_image_generator = ImageDataGenerator(\n",
" rescale=1./255\n",
")"
"val_image_generator = ImageDataGenerator( rescale=1./255 )"
]
},
{
@ -572,7 +548,7 @@
"metadata": {},
"outputs": [],
"source": [
"model.save('/srv/satnogs/data/models/PSK31/wut-PSK31-201912.h5')"
"model.save('/srv/satnogs/data/models/GMSK/wut-GMSK-201912.h5')"
]
},
{
@ -581,7 +557,7 @@
"metadata": {},
"outputs": [],
"source": [
"model.save('/srv/satnogs/data/models/PSK31/wut-PSK31-201912.tf')"
"model.save('/srv/satnogs/data/models/GMSK/wut-GMSK-201912.tf')"
]
},
{