# CY

**Repository Path**: majx1997/cy

## Basic Information

- **Project Name**: CY
- **Description**: 吸烟打电话检测
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 2
- **Created**: 2020-09-25
- **Last Updated**: 2024-12-07

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 吸烟打电话检测比赛
  
1. 主函数在efficient.py

2. 目前使用的是efficient-b5模型，CrossEntropyLoss损失函数，余弦退火学习率衰减策略， 线上最好得训练图像大小是600*600， 增广使用了模糊/对比度/水平翻转

3. 通过修改`config/efficient.yaml`配置可以控制模型训练参数， 训练图像大小需要通过在`utils/dataset.py`里面调整

4. 推理代码是`infer.py`, 训练代码是`engine/trainer.py`，通过`script/convert_data.py`划分了训练集和验证集(0.1比例)