# 写在前面
+ 本项目所有代码基于paddle的notebook(AI Stidio经典版)运行。
# 项目背景
+ 飞桨领航团AI达人创造营
# 数据集
+ [绵羊品种分类](https://aistudio.baidu.com/aistudio/datasetdetail/108109)
# 模型介绍
```python
num_classes = len(train_dataset.labels)
model = pdx.cls.MobileNetV2(num_classes=num_classes)
```
# 模型训练
```python
model.train(num_epochs=100,
train_dataset=train_dataset,
train_batch_size=64,
eval_dataset=val_dataset,
lr_decay_epochs=[30, 60, 90],
save_interval_epochs=1,
pretrain_weights='IMAGENET',
learning_rate=0.0003,
save_dir='output/mobilenetv2',
use_vdl=True)
```
# 结果验证
```python
test_transforms = T.Compose(
[T.ResizeByShort(short_size=256), T.CenterCrop(crop_size=224), T.Normalize()])
model = pdx.load_model('output/mobilenetv2/best_model')
image_name = 'work/SheepFaceImages/Poll_Dorset/000363_P.jpg'
result = model.predict(image_name,test_transforms)
print("Predict Result:", result)
```
# 总结
+ paddle作为一款国产深度学习框架,对比TF、pytouch等其他框架来说,资料相对好找,作为入门Deep Learning是一个很好的选择。
# 个人主页
+ [CSDN](https://blog.csdn.net/bblood307?type=blog)
+ [github](https://github.com/bbloodtech)
+ [gitee](https://gitee.com/bbloodtech)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
基于paddle的绵羊品种分类源码+项目说明+数据集.zip # 数据集 + [绵羊品种分类](https://aistudio.baidu.com/aistudio/datasetdetail/108109) # 模型介绍 ```python num_classes = len(train_dataset.labels) model = pdx.cls.MobileNetV2(num_classes=num_classes) ``` # 模型训练 ```python model.train(num_epochs=100, train_dataset=train_dataset, train_batch_size=64, eval_dataset=val_dataset, lr_decay_epochs=[30, 60, 90], save_interval_epochs=1, pretrain_weights='IMAGENET', learning_rate=0.0003, save_dir='output/mobilenetv2', use_vdl=True) ``` # 结果验
资源推荐
资源详情
资源评论
收起资源包目录
基于paddle的绵羊品种分类源码+项目说明+数据集.zip (2个子文件)
code
main.ipynb 508KB
README.md 1KB
共 2 条
- 1
资源评论
FL1768317420
- 粉丝: 4590
- 资源: 4813
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功