Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -54,14 +54,14 @@
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": 5,
"id": "734f7a66",
"metadata": {},
"outputs": [],
"source": [
"from datamint import Api\n",
"\n",
"PROJECT_NAME = \"FracAtlas\"\n",
"PROJECT_NAME = \"ClassificationTest\"\n",
"api = Api()"
]
},
Expand Down Expand Up @@ -106,10 +106,29 @@
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 8,
"id": "bb794bc8",
"metadata": {},
"outputs": [],
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"INFO:datamint.dataset.factory:Detected resource type 'image'; using ImageDataset.\n",
"INFO:datamint.dataset.base:Allowing external image label '('has_fracture', 'no')' not in project specs.\n",
"INFO:datamint.dataset.base:Allowing external image label '('has_fracture', 'yes')' not in project specs.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"train : 3268 resources\n",
"val : 408 resources\n",
"test : 409 resources\n"
]
}
],
"source": [
"from datamint.dataset import build_dataset\n",
"\n",
Expand All @@ -121,7 +140,7 @@
")\n",
"\n",
"parts = dataset.split(train=0.8, val=0.1, test=0.1, seed=123)\n",
"parts.save() \n",
"parts.save(force=True) \n",
"\n",
"for name, ds in parts.items():\n",
" print(f'{name:6s}: {len(ds):4d} resources')"
Expand Down Expand Up @@ -165,8 +184,9 @@
" project=PROJECT_NAME,\n",
" image_size=384,\n",
" batch_size=8,\n",
" max_epochs=10,\n",
" max_epochs=3,\n",
" accelerator='auto',\n",
" pretrained=False,\n",
" # model_name='MyModelName' # defaults to project name\n",
")\n",
"\n",
Expand Down
Loading