/pytorch_pix2code

A pytorch Implementation of pix2code

Primary LanguageJupyter Notebook

Pix2Code pytorch 版

以下のURLの実装です

https://vaibhavyadav.github.io/2020/05/20/ml-challenge-implementing-pix2code-in-pytorch.html

変更点

  • Pytorch (Jupyter) から Python へ変更

実行方法

python3 ./pix2code.py

データセットについて

  • dataset/ に学習用・評価用に分類したデータが格納されています。

実行結果

Loss: 0.11163423210382462, Epoch: 9
Loss: 0.09132043272256851, Epoch: 9
Loss: 0.08423051238059998, Epoch: 9
Loss: 0.08951333910226822, Epoch: 9
Loss: 0.10713265836238861, Epoch: 9
Loss: 0.08967424929141998, Epoch: 9
Loss: 0.092195525765419, Epoch: 9
Loss: 0.0718386098742485, Epoch: 9
Loss: 0.08632747828960419, Epoch: 9
Loss: 0.08393169194459915, Epoch: 9
Loss: 0.08250757306814194, Epoch: 9
Loss: 0.09406688064336777, Epoch: 9
Loss: 0.09367651492357254, Epoch: 9
Loss: 0.086444191634655, Epoch: 9
Loss: 0.09337592124938965, Epoch: 9
Loss: 0.09244504570960999, Epoch: 9
Loss: 0.09103869646787643, Epoch: 9
Loss: 0.08954822272062302, Epoch: 9
Loss: 0.09800052642822266, Epoch: 9
Loss: 0.08404076099395752, Epoch: 9
Loss: 0.08680231124162674, Epoch: 9
Loss: 0.08986985683441162, Epoch: 9
Loss: 0.09012219309806824, Epoch: 9
Loss: 0.08973389118909836, Epoch: 9
Loss: 0.08540251851081848, Epoch: 9
Loss: 0.08377905189990997, Epoch: 9
Loss: 0.07200120389461517, Epoch: 9
header{
btn-inactive,btn-inactive,btn-inactive,btn-inactive
}
row{
double{
small-title,text,btn-red
}
double{
small-title,text,btn-green
}
}