提交 d6439b6d 编写于 作者: F FlyingQianMM

change python3 to python

上级 ee31e819
...@@ -90,19 +90,19 @@ export CUDA_VISIBLE_DEVICES= ...@@ -90,19 +90,19 @@ export CUDA_VISIBLE_DEVICES=
* 预测单张图片 * 预测单张图片
```shell ```shell
python3 reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image /path/to/meter_test/20190822_168.jpg --save_dir ./output --use_erode python reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image /path/to/meter_test/20190822_168.jpg --save_dir ./output --use_erode
``` ```
* 预测多张图片 * 预测多张图片
```shell ```shell
python3 reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image_dir /path/to/meter_test --save_dir ./output --use_erode python reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image_dir /path/to/meter_test --save_dir ./output --use_erode
``` ```
* 开启摄像头预测 * 开启摄像头预测
```shell ```shell
python3 reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --save_dir ./output --use_erode --use_camera python reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --save_dir ./output --use_erode --use_camera
``` ```
## 推理部署 ## 推理部署
...@@ -259,12 +259,12 @@ step 5. 推理预测: ...@@ -259,12 +259,12 @@ step 5. 推理预测:
* 表盘检测的训练 * 表盘检测的训练
``` ```
python3 /path/to/PaddleX/examples/meter_reader/train_detection.py python /path/to/PaddleX/examples/meter_reader/train_detection.py
``` ```
* 指针和刻度分割的训练 * 指针和刻度分割的训练
``` ```
python3 /path/to/PaddleX/examples/meter_reader/train_segmentation.py python /path/to/PaddleX/examples/meter_reader/train_segmentation.py
``` ```
......
...@@ -31,14 +31,14 @@ cd PaddleX/examples/remote_sensing/ ...@@ -31,14 +31,14 @@ cd PaddleX/examples/remote_sensing/
运行以下脚本,下载原始数据集,并完成数据集的切分: 运行以下脚本,下载原始数据集,并完成数据集的切分:
``` ```
python3 prepare_data.py python prepare_data.py
``` ```
## 模型训练 ## 模型训练
分割模型选择Backbone为MobileNetv3_large_ssld的Deeplabv3模型,该模型兼备高性能高精度的优点。运行以下脚本,进行模型训练: 分割模型选择Backbone为MobileNetv3_large_ssld的Deeplabv3模型,该模型兼备高性能高精度的优点。运行以下脚本,进行模型训练:
``` ```
python3 train.py python train.py
``` ```
也可以跳过模型训练步骤,直接下载预训练模型进行后续的模型预测和评估: 也可以跳过模型训练步骤,直接下载预训练模型进行后续的模型预测和评估:
...@@ -71,12 +71,12 @@ tar -xvf ccf_remote_model.tar.gz ...@@ -71,12 +71,12 @@ tar -xvf ccf_remote_model.tar.gz
运行以下脚本使用有重叠的滑动窗口进行预测: 运行以下脚本使用有重叠的滑动窗口进行预测:
``` ```
python3 predict.py python predict.py
``` ```
## 模型评估 ## 模型评估
在训练过程中,每隔10个迭代轮数会评估一次模型在验证集的精度。由于已事先将原始大尺寸图片切分成小块,此时相当于使用无重叠的大图切小图预测方式,最优模型精度miou为80.58%。运行以下脚本,将采用有重叠的大图切小图的预测方式,重新评估原始大尺寸图片的模型精度,此时miou为81.52%。 在训练过程中,每隔10个迭代轮数会评估一次模型在验证集的精度。由于已事先将原始大尺寸图片切分成小块,此时相当于使用无重叠的大图切小图预测方式,最优模型精度miou为80.58%。运行以下脚本,将采用有重叠的大图切小图的预测方式,重新评估原始大尺寸图片的模型精度,此时miou为81.52%。
``` ```
python3 eval.py python eval.py
``` ```
...@@ -99,19 +99,19 @@ export CUDA_VISIBLE_DEVICES= ...@@ -99,19 +99,19 @@ export CUDA_VISIBLE_DEVICES=
* 预测单张图片 * 预测单张图片
```shell ```shell
python3 reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image /path/to/meter_test/20190822_168.jpg --save_dir ./output --use_erode python reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image /path/to/meter_test/20190822_168.jpg --save_dir ./output --use_erode
``` ```
* 预测多张图片 * 预测多张图片
```shell ```shell
python3 reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image_dir /path/to/meter_test --save_dir ./output --use_erode python reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --image_dir /path/to/meter_test --save_dir ./output --use_erode
``` ```
* 开启摄像头预测 * 开启摄像头预测
```shell ```shell
python3 reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --save_dir ./output --use_erode --use_camera python reader_infer.py --detector_dir /path/to/det_inference_model --segmenter_dir /path/to/seg_inference_model --save_dir ./output --use_erode --use_camera
``` ```
## <h2 id="4">推理部署</h2> ## <h2 id="4">推理部署</h2>
...@@ -269,12 +269,12 @@ git clone https://github.com/PaddlePaddle/PaddleX ...@@ -269,12 +269,12 @@ git clone https://github.com/PaddlePaddle/PaddleX
* 表盘检测的训练 * 表盘检测的训练
``` ```
python3 /path/to/PaddleX/examples/meter_reader/train_detection.py python /path/to/PaddleX/examples/meter_reader/train_detection.py
``` ```
* 指针和刻度分割的训练 * 指针和刻度分割的训练
``` ```
python3 /path/to/PaddleX/examples/meter_reader/train_segmentation.py python /path/to/PaddleX/examples/meter_reader/train_segmentation.py
``` ```
......
...@@ -3,7 +3,7 @@ from PIL import Image ...@@ -3,7 +3,7 @@ from PIL import Image
import paddlex as pdx import paddlex as pdx
model_dir = "saved_model/remote_sensing_unet/best_model/" model_dir = "l8sparcs_remote_model/"
img_file = "dataset/remote_sensing_seg/data/LC80150242014146LGN00_23_data.tif" img_file = "dataset/remote_sensing_seg/data/LC80150242014146LGN00_23_data.tif"
label_file = "dataset/remote_sensing_seg/mask/LC80150242014146LGN00_23_mask.png" label_file = "dataset/remote_sensing_seg/mask/LC80150242014146LGN00_23_mask.png"
color = [255, 255, 255, 0, 0, 0, 255, 255, 0, 255, 0, 0, 150, 150, 150] color = [255, 255, 255, 0, 0, 0, 255, 255, 0, 255, 0, 0, 150, 150, 150]
...@@ -11,6 +11,7 @@ color = [255, 255, 255, 0, 0, 0, 255, 255, 0, 255, 0, 0, 150, 150, 150] ...@@ -11,6 +11,7 @@ color = [255, 255, 255, 0, 0, 0, 255, 255, 0, 255, 0, 0, 150, 150, 150]
# 预测并可视化预测结果 # 预测并可视化预测结果
model = pdx.load_model(model_dir) model = pdx.load_model(model_dir)
pred = model.predict(img_file) pred = model.predict(img_file)
#pred = model.overlap_tile_predict(img_file, tile_size=[512, 512], pad_size=[64, 64], batch_size=32)
pdx.seg.visualize( pdx.seg.visualize(
img_file, pred, weight=0., save_dir='./output/pred', color=color) img_file, pred, weight=0., save_dir='./output/pred', color=color)
......
...@@ -37,14 +37,14 @@ cd PaddleX/examples/remote_sensing/ ...@@ -37,14 +37,14 @@ cd PaddleX/examples/remote_sensing/
运行以下脚本,下载原始数据集,并完成数据集的切分: 运行以下脚本,下载原始数据集,并完成数据集的切分:
``` ```
python3 prepare_data.py python prepare_data.py
``` ```
## <h2 id="2">模型训练</h2> ## <h2 id="2">模型训练</h2>
分割模型选择Backbone为MobileNetv3_large_ssld的Deeplabv3模型,该模型兼备高性能高精度的优点。运行以下脚本,进行模型训练: 分割模型选择Backbone为MobileNetv3_large_ssld的Deeplabv3模型,该模型兼备高性能高精度的优点。运行以下脚本,进行模型训练:
``` ```
python3 train.py python train.py
``` ```
也可以跳过模型训练步骤,直接下载预训练模型进行后续的模型预测和评估: 也可以跳过模型训练步骤,直接下载预训练模型进行后续的模型预测和评估:
...@@ -77,12 +77,12 @@ tar -xvf ccf_remote_model.tar.gz ...@@ -77,12 +77,12 @@ tar -xvf ccf_remote_model.tar.gz
运行以下脚本使用有重叠的滑动窗口进行预测: 运行以下脚本使用有重叠的滑动窗口进行预测:
``` ```
python3 predict.py python predict.py
``` ```
## <h2 id="4">模型评估</h2> ## <h2 id="4">模型评估</h2>
在训练过程中,每隔10个迭代轮数会评估一次模型在验证集的精度。由于已事先将原始大尺寸图片切分成小块,此时相当于使用无重叠的滑动窗口预测方式,最优模型精度miou为80.58%。运行以下脚本,将采用有重叠的滑动窗口预测方式,重新评估原始大尺寸图片的模型精度,此时miou为81.52%。 在训练过程中,每隔10个迭代轮数会评估一次模型在验证集的精度。由于已事先将原始大尺寸图片切分成小块,此时相当于使用无重叠的滑动窗口预测方式,最优模型精度miou为80.58%。运行以下脚本,将采用有重叠的滑动窗口预测方式,重新评估原始大尺寸图片的模型精度,此时miou为81.52%。
``` ```
python3 eval.py python eval.py
``` ```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册