# road_classification_v2

**Repository Path**: max__niu/road_classification_v2

## Basic Information

- **Project Name**: road_classification_v2
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2025-07-09
- **Last Updated**: 2025-07-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 路况分类模型


## 1.准备数据集



## 2.数据集转txt文件

```
python gen_data_txt_file.py --data_path terrains1
```

## 3.修改对应配置

arsg.py

```
dataset_name = 'terrains1'
```

## 4.训练

```
python main.py --mode train
```

## 5.pth测试

```
python main.py --mode test
```

## 6.转onnx

```
python get_onnx_om.py --mode test
```

## 7.onnx测试

```
python test_onnx.py --mode test --batch_size 1
```

## 8.转om

```
atc --framework=5 \
    --model=../terrains1_bs1_224.onnx \
    --output=../terrains1_bs1_224_aipp \
    --input_format=NCHW \
    --input_shape="image:1,3,224,224" \
    --log=error \
    --soc_version=Ascend310B1 \
    --enable_small_channel=1 \
    --insert_op_conf=aipp_resnet50.config
```

## 9.om测试

```
python test_om.py --mode test --batch_size 1
```