# 自动打拳机器人

**Repository Path**: cangyeone/automatic-boxing-robot

## Basic Information

- **Project Name**: 自动打拳机器人
- **Description**: 全自动打拳机器人，基于PyTorch构建的RNN完成。
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 2
- **Forks**: 0
- **Created**: 2021-11-18
- **Last Updated**: 2024-06-26

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 自动打拳机器人

#### 介绍
全自动打拳机器人，基于PyTorch构建的RNN完成。

#### 说明
完全基于PyTorch构建，并不需要其他乱七八糟的东西。预训练模型在ckpt文件夹写下，由于训练数据较少模型处于过拟合状态。后续可以自行添加数据。


#### 训练

训练数据在data文件夹下的mad。直接运行
```bash
python 女拳模拟器-train.py
```
#### 预测

文件中定义开始词语的部分在
```python 
model.load_state_dict(torch.load("ckpt/mad.pt"))
pres = ['男','女','爱']model.load_state_dict(torch.load("ckpt/mad.pt"))
pres = ['男','女','爱']
```
之后运行

```bash
python 女拳模拟器-valid.py
```
#### 参与贡献

如是，cangye@Hotmail.com

#### LICENCE 
MIT