Giter Club home page Giter Club logo

ugtr's Introduction

Uncertainty-Guided Transformer Reasoning for Camouflaged Object Detection (ICCV2021)

Authors: Fan Yang, Qiang Zhai, Xin Li, Rui Huang, Hong Cheng, Deng-Ping Fan.

  1. Configuring your environment (Prerequisites):

    Pytorch>=1.0.0

    OpenCV

  1. Downloading Testing Sets:

    • downloading NEW testing dataset (COD10K-test + CAMO-test + CHAMELEON), which can be found in this Google Drive link or Baidu Pan link with the fetch code: z83z.
  2. Testing Configuration:

    • After you download the trained models Google Drive link or Baidu Pan link, move it into './model_file/'.
    • Assigning your comstomed path in 'config/cod_resnet50.yaml', like 'data_root', 'test_list'.
    • Playing 'test.py' to generate the final prediction map, the predicted camouflaged object region and cmouflaged object edge is saved into 'result' as default.
  3. Evaluation your trained model:

    • One-key evaluation is written in MATLAB code (revised from link), please follow this the instructions in main.m and just run it to generate the evaluation results in ./EvaluationTool/EvaluationResults/Result-CamObjDet/.
    • The results can be downloaded in Baidu Pan link(password: 2kj3).

  1. Training Configuration:

    • After you download the initial model Google Drive link or Baidu Pan link, move it to './pre_trained/'.
    • Put the 'train_test_file/train.lst' to the path which is included in cod_resnet50.yaml.
    • Run train.py
  2. If you think this work is helpful, please cite

@inproceedings{fan2021ugtr,
  title={Uncertainty-Guided Transformer Reasoning for Camouflaged Object Detection},
  author={Yang, Fan and Zhai, Qiang and Li, Xin and Huang, Rui and Cheng, Hong and Fan, Deng-Ping},
  booktitle={IEEE International Conference on Computer Vision(ICCV)},
  pages={},
  year={2021}
}

ugtr's People

Contributors

fanyang587 avatar

Stargazers

Xinhui Lin avatar  avatar WBC-ML avatar Lucas avatar yuxin chen avatar Wang Jia Huan avatar a-rui avatar  avatar Hungsing avatar  avatar  avatar  avatar winterswee avatar leo1 avatar  avatar Ge Wu avatar fan avatar  avatar Khoa Nguyen avatar Wang Xiaohang avatar Xiaobin HU(kevin) avatar  avatar Ellery Queen avatar  avatar Siyuan Yan avatar Vermillion avatar  avatar Abe avatar An-zhi WANG avatar Yang Yang avatar

Watchers

James Cloos avatar  avatar

ugtr's Issues

I think there's some issues on the ugtr.py

  1. After line #97, if prob_x should be interpreted as probability I think sigmoid function should be applied to prob_x as well as prob_out2.

  2. I haven't run file train.py (since the training data is not given...), but if target is binary consist of only 1 and 0 (it seems like you are using torch.where and divide the target by 255.0 to make it binary, in train.py), KL divergence loss always gives negative number (Please refer to the pytorch document).

I figured this issues while applying your method to other dataset, so I might be wrong.
Please reply to this issue.

Thanks.

Joseph

显存12G,batch_size只能是1

尊敬的作者,我尝试用你们的网络训练自己的数据集,发现代码里默认的batch_size是20,但是由于我的显卡内存才是1G,只能支持1个batch_size,请问这是一个正常情况嘛,想问问作者您在训练的时候batch_size也是20嘛。

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.