大海

SIMPLE THE COMPLEX

阅读源码时发现torchvision提供了许多边界框操作,包括

  1. NMS
  2. 移除小边界框
  3. 修剪边界框坐标,保证边界框坐标位于图像内
  4. 计算边界框面积
  5. IoU计算

在线文档:torchvision.ops

源码地址:torchvision/ops/boxes.py

当前torchvision版本:0.6.0a0+82fd1c8

阅读全文 »

4月份实现了YOLO_v1算法,学习新的模型ResNet/GoogLeNet/SqueezeNet/DenseNet/SSD,同时小结了评估标准,包括mAP/Flops/Params Size/FPS/Top-k accuracy

本周学习了如何寻找最优学习率/权重衰减,以及进行warmup+CosineAnnearling加速模型训练,同时加强了数据预处理,包括随机擦除/颜色抖动/Ten Crops测试等

本周继续SSD算法的学习和实现,同时开始新模型和训练方法的学习。本月期待完成新工作的寻找

阅读全文 »