未验证 提交 c19a4b26 编写于 作者: G guo ran 提交者: GitHub

Merge pull request #18 from Oneflow-Inc/build_process_update

add darknet submodule, update build.sh and readme
[submodule "third_party/darknet"]
path = third_party/darknet
url = https://github.com/pjreddie/darknet
......@@ -22,7 +22,7 @@
1.git clone[此仓库](https://github.com/Oneflow-Inc/oneflow_yolov3)到本地
```shell
git clone https://github.com/Oneflow-Inc/oneflow_yolov3.git
git clone --recursive https://github.com/Oneflow-Inc/oneflow_yolov3.git
```
2.安装python依赖库
......@@ -34,7 +34,7 @@ git clone https://github.com/Oneflow-Inc/oneflow_yolov3.git
3.在项目root目录下,执行:
```
bash scripts/build.sh
./scripts/build.sh
```
执行此脚本,将cpp代码中自定义的op算子编译成可调用执行的.so文件,您将在项目路径下看到:
......
......@@ -38,7 +38,7 @@ target_include_directories(${PROJECT_NAME} PUBLIC ${ONEFLOW_INCLUDE_PATH})
target_link_libraries(${PROJECT_NAME} ${ONEFLOW_LIB_PATH}/_oneflow_internal.so)
target_include_directories(${PROJECT_NAME} PUBLIC ${CMAKE_SOURCE_DIR}/ops/darknet_dependencies/include)
target_link_libraries(${PROJECT_NAME} ${CMAKE_SOURCE_DIR}/oneflow_yolov3/libdarknet.so)
target_link_libraries(${PROJECT_NAME} ${CMAKE_SOURCE_DIR}/third_party/darknet/libdarknet.so)
export(PACKAGE ${PROJECT_NAME})
......
#!/usr/bin/env bash
set -e
set -x
bash scripts/clean.sh
export ONEFLOW_DEBUG_MODE=""
mkdir build && cd build
cmake .. && make -j
cd ..
pushd third_party/darknet
sed -i '1s/0/1/' Makefile
make -j`nproc`
popd
mkdir -p build
pushd build
cmake ..
make -j`nproc`
popd
pip3 install -e . --user
#!/usr/bin/env bash
set -x
rm -rf oneflow_yolov3/liboneflow_yolov3.so
rm -rf build
......
#!/bin/bash
#!/usr/bin/env bash
# # Download Images
# wget -c https://pjreddie.com/media/files/train2014.zip
# wget -c https://pjreddie.com/media/files/val2014.zip
......@@ -12,4 +13,4 @@ paste <(awk "{print \"$PWD\"}" <trainvalno5k.part) trainvalno5k.part | tr -d '\t
# copy label txt to image dir
find labels/train2014/ -name "*.txt" | xargs -i cp {} images/train2014/
find labels/val2014/ -name "*.txt" | xargs -i cp {} images/val2014/
\ No newline at end of file
find labels/val2014/ -name "*.txt" | xargs -i cp {} images/val2014/
#!/usr/bin/env bash
set -x
pip3 install -e . --user --verbose
#!/usr/bin/env bash
set -x
OF_CFLAGS=( $(python3 -c 'import oneflow; print(" ".join(oneflow.sysconfig.get_compile_flags()))') )
OF_LFLAGS=( $(python3 -c 'import oneflow; print(" ".join(oneflow.sysconfig.get_link_flags()))') )
......
Subproject commit bc5ea7cca8c770ebdacd2c792ddd96f6904ee23a
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册