Giter Club home page Giter Club logo

unet-paddle-daheng's Introduction

Unet-paddle-daheng

Daheng dataset Segmentation Task implement on paddlepaddle

.
|-- Unet-paddle
|   |-- daheng_best_model
|   |   `-- best_end
|   |-- daheng_test.py
|   |-- daheng_train.py
|   |-- data
|   |   |-- __init__.py
|   |   |-- data_feeder.py
|   |   `-- reader.py
|   |-- model
|   |   |-- __init__.py
|   |   `-- unet.py
|   |-- test.sh
|   `-- train.sh
`-- daheng
    |-- images
    |   |-- 其他
    |   |-- 异物
    |   |-- 漏涂
    |   |-- 烧伤
    |   |-- 白点
    |   |-- 黑点
    |   |-- 小白线
    |   `-- 孔或透明
    |-- labels
    |   |-- 其他
    |   |-- 异物
    |   |-- 漏涂
    |   |-- 烧伤
    |   |-- 白点
    |   |-- 黑点
    |   |-- 小白线
    |   `-- 孔或透明
    |-- test_list.txt
    `-- train_list.txt

Train

CUDA_VISIBLE_DEVICES=0,1,2 sh train.sh

python ./daheng_train.py
-train_file ./daheng/train_list.txt
-valid_file ./test_list.txt
-save_model ./daheng_model (save model param path)
-save_infer ./daheng_infer (save infer model path)
-pretrain_model ./best_end (pretrain model path)
-data_dir ./daheng/

Test

CUDA_VISIBLE_DEVICES=0 sh test.sh

python ./daheng_test.py
-test_file ./daheng/test_list.txt
-model_path ./best_end (best model path for predict)
-data_dir ./daheng/
-pre_savedir ./save_pre_image/

unet-paddle-daheng's People

Contributors

yanyuetman avatar

Stargazers

avatar

Watchers

avatar

Forkers

chunda-zhang

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.