PyTorch中的RandomErasing实现：pytorch_random_erasing
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
简介：pytorch_random_erasing是基于PyTorch框架的一个随机擦除数据增强工具包，用于图像处理任务中提高模型泛化能力。
PyTorch随机擦除的实现用法如下：
```python
$ python main.py --block_type basic --depth 110 --use_random_erase --random_erase_prob 0.5 --random_erase_area_ratio_range [0.02, 0.4] --random_erase_min_aspect_ratio 0.3 --random_erase_max_attempt 20 --outdir results
```
CIFAR-10的结果模型测试错误（基于五次运行的中位数）如下：
- 没有随机擦除的ResNet-preact-56：5.85，训练时间98分钟
- 使用随机擦除的ResNet-preact-56：5.22，训练时间同样为98分钟
另外一种配置示例如下：
```python
$ python -u main.py --depth 56 --block_type b
```
以上是PyTorch中实现随机擦除的具体命令和使用效果展示。
全部评论 (0)
还没有任何评论哟~
