Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
spacesick committed Jun 12, 2023
1 parent 65b1e99 commit 1fb7db6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ We collected our Dataset various wound type from kaggle. We got 431 wound type i

### Architecture

We use the technique of transfer learning by adding multiple layers of fully connected networks on top of an InceptionV3 model pre-trained on the imagenet dataset. We freeze all the layers in the pre-trained InceptionV3 except for the last 12. We saved our best performing model in .h5 format and its weights in a .ckpt file, which so far has reached a validation accuracy of 0.8315. The latest checkpoints can be found [here](https://drive.google.com/drive/folders/1JRxoLPASzwMdFGjUG7gDhSLfg6zoRx-t?usp=sharing). History of build/develop model before fix model can be found [here](https://github.com/dhuhaardha/Model_Capstone)
We use the technique of transfer learning by adding multiple layers of fully connected networks on top of an InceptionV3 model pre-trained on the imagenet dataset. We freeze all the layers in the pre-trained InceptionV3 except for the last 12. We saved our best performing model in .h5 format and its weights in a .ckpt file, which so far has reached a validation accuracy of over 0.83 for the injury classification task and over 0.89 for the accident classification task. The dataset, latest checkpoints, and various model formats can be found [here](https://drive.google.com/drive/folders/1JRxoLPASzwMdFGjUG7gDhSLfg6zoRx-t?usp=sharing). Legacy versions of the model can be found [here](https://github.com/dhuhaardha/Model_Capstone).

## Cloud computing and deployment

Expand Down

0 comments on commit 1fb7db6

Please sign in to comment.