From 143e22d847da0732c2783ef0f16d14bf63b72e62 Mon Sep 17 00:00:00 2001
From: cnn <liuhui29@baidu.com>
Date: Fri, 16 Apr 2021 14:01:16 +0800
Subject: [PATCH] fix bug of import and update doc (#2643)

---
 configs/dota/README.md                  | 9 ++-------
 ppdet/data/transform/batch_operators.py | 1 +
 ppdet/data/transform/operators.py       | 1 +
 3 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/configs/dota/README.md b/configs/dota/README.md
index 07521d195..250d0bc9a 100644
--- a/configs/dota/README.md
+++ b/configs/dota/README.md
@@ -54,11 +54,6 @@ DOTA数据集中总共有2806张图像，其中1411张图像作为训练集，45
 sudo nvidia-docker run -it --name paddle_s2anet -v $PWD:/paddle --network=host registry.baidubce.com/paddlepaddle/paddle:2.0.1-gpu-cuda10.1-cudnn7 /bin/bash
 ```
 
-进入容器后，安装必要的python包：
-```
-python3.7 -m pip install Cython wheel tqdm opencv-python==4.2.0.32 scipy  PyYAML shapely pycocotools
-```
-
 镜像中paddle2.0.1已安装好，进入python3.7，执行如下代码检查paddle安装是否正常：
 ```
 import paddle
@@ -66,7 +61,7 @@ print(paddle.__version__)
 paddle.utils.run_check()
 ```
 
-进入到`ext_op`文件夹，安装：
+进入到`ppdet/ext_op`文件夹，安装：
 ```
 python3.7 setup.py install
 ```
@@ -96,7 +91,7 @@ Paddle中`multiclass_nms`算子的输入支持四边形输入，因此部署时
 
 ```bash
 # 预测
-CUDA_VISIBLE_DEVICES=0 python tools/infer.py -c configs/dota/s2anet_1x_dota.yml -o weights=model.pdparams --infer_img=demo/P0072__1.0__0___0.png --use_gpu=True
+CUDA_VISIBLE_DEVICES=0 python tools/infer.py -c configs/dota/s2anet_1x_dota.yml -o weights=model.pdparams --infer_img=demo/P0072__1.0__0___0.png
 ```
 
 
diff --git a/ppdet/data/transform/batch_operators.py b/ppdet/data/transform/batch_operators.py
index e11c48980..85639c31d 100644
--- a/ppdet/data/transform/batch_operators.py
+++ b/ppdet/data/transform/batch_operators.py
@@ -27,6 +27,7 @@ from .operators import register_op, BaseOperator, Resize
 from .op_helper import jaccard_overlap, gaussian2D
 from scipy import ndimage
 
+from ppdet.modeling import bbox_utils
 from ppdet.utils.logger import setup_logger
 logger = setup_logger(__name__)
 
diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index 65608f367..932c7971f 100644
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -39,6 +39,7 @@ from PIL import Image, ImageEnhance, ImageDraw
 
 from ppdet.core.workspace import serializable
 from ppdet.modeling.layers import AnchorGrid
+from ppdet.modeling import bbox_utils
 
 from .op_helper import (satisfy_sample_constraint, filter_and_process,
                         generate_sample_bbox, clip_bbox, data_anchor_sampling,
-- 
GitLab