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

Update README.md

parent 2b27ace2
# 1. 环境准备
docker pull image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.18.0-ubuntu22.04-dtk25.04.2-py3.10
# ParticleNet
## 项目简介
# 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 \
--name ParticleNet \
--privileged \
......@@ -15,19 +22,30 @@ docker run -it \
--security-opt seccomp=unconfined \
image.sourcefind.cn:5000/dcu/admin/base/tensorflow:2.18.0-ubuntu22.04-dtk25.04.2-py3.10 \
/bin/bash
```
---
## 测试步骤
### 1. 拉取代码
# 3. 拉取代码&安装依赖
```bash
cd /workspace
git clone http://developer.sourcefind.cn/codes/modelzoo/particlenet_tensorflow.git
cd particlenet_tensorflow/tf-keras
```
### 2. 安装依赖
```bash
pip install -r requirements.txt
pip install tf_keras==2.18.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --no-deps
```
### 3. 下载数据集
# 4. 下载&预处理数据集
```bash
mkdir original
cd original
......@@ -38,8 +56,10 @@ curl -O -L https://zenodo.org/record/2603256/files/val.h5
cd ..
python convert_dataset.py # 程序结束后生成converted目录
```
# 5. 测试命令
### 4. 执行测试命令
```bash
训练:python keras_train.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