Skip to content

Commit

Permalink
delete wrong description.
Browse files Browse the repository at this point in the history
  • Loading branch information
xiaochus committed May 21, 2019
1 parent d147523 commit f881764
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 17 deletions.
19 changes: 2 additions & 17 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ According to the paper: [Inverted Residuals and Linear Bottlenecks Mobile Networ
## Requirement
- OpenCV 3.4
- Python 3.5
- Tensorflow-gpu 1.2.0
- Keras 2.1.3
- Tensorflow-gpu 1.5.0
- Keras 2.2


## MobileNet v2 and inverted residual block architectures
Expand Down Expand Up @@ -65,22 +65,7 @@ python train.py --classes num_classes --batch batch_size --epochs epochs --size
- --weights, Fine tune with other weights.
- --tclasses, The number of classes of pre-trained model.

## Experiment

Due to the limited computational resources, we used cifar-100 dataset to test the model.

device: Tesla K80
dataset: cifar-100
optimizer: Adam(lr=0.001, beta_1=0.9, beta_2=0.999, epsilon=1e-08)
batch_szie: 128

These are the details for the CIFAR-100 experiment. Although the network did not completely converge, still achieved good accuracy.

| Metrics | Loss | Top-1 Accuracy | Top-5 Accuracy |
| ------- |------| :------------: | :------------: |
| cifar-100 | 0.195 | 94.42% | 99.82% |

![evaluate](/images/eva.png)
## Reference

@article{MobileNetv2,
Expand Down
Binary file removed images/eva.png
Binary file not shown.

0 comments on commit f881764

Please sign in to comment.