Commit 35f6569d authored by zhangwenbo's avatar zhangwenbo 🎨
Browse files

Update README.md

parent 2b27ace2
# 1. 环境准备 # ParticleNet
docker pull image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.18.0-ubuntu22.04-dtk25.04.2-py3.10
## 项目简介
# 2. 创建容器 ## 环境部署
### 1. 拉取镜像
```bash
docker pull image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.18.0-ubuntu22.04-dtk25.04.2-py3.10
```
### 2. 创建容器
```bash
docker run -it \ docker run -it \
--name ParticleNet \ --name ParticleNet \
--privileged \ --privileged \
...@@ -15,19 +22,30 @@ docker run -it \ ...@@ -15,19 +22,30 @@ docker run -it \
--security-opt seccomp=unconfined \ --security-opt seccomp=unconfined \
image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.18.0-ubuntu22.04-dtk25.04.2-py3.10 \ image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.18.0-ubuntu22.04-dtk25.04.2-py3.10 \
/bin/bash /bin/bash
```
---
## 测试步骤
### 1. 拉取代码
# 3. 拉取代码&安装依赖 ```bash
cd /workspace cd /workspace
git clone http://developer.sourcefind.cn/codes/modelzoo/particlenet_tensorflow.git git clone http://developer.sourcefind.cn/codes/modelzoo/particlenet_tensorflow.git
cd particlenet_tensorflow/tf-keras cd particlenet_tensorflow/tf-keras
```
### 2. 安装依赖
```bash
pip install -r requirements.txt pip install -r requirements.txt
pip install tf_keras==2.18.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --no-deps pip install tf_keras==2.18.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --no-deps
```
### 3. 下载数据集
# 4. 下载&预处理数据集
```bash
mkdir original mkdir original
cd original cd original
...@@ -38,8 +56,10 @@ curl -O -L https://zenodo.org/record/2603256/files/val.h5 ...@@ -38,8 +56,10 @@ curl -O -L https://zenodo.org/record/2603256/files/val.h5
cd .. cd ..
python convert_dataset.py # 程序结束后生成converted目录 python convert_dataset.py # 程序结束后生成converted目录
```
# 5. 测试命令 ### 4. 执行测试命令
```bash
训练:python keras_train.py 训练:python keras_train.py
推理:python predict_demo.py 推理:python predict_demo.py
```
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment