/context-encoder

Implement context encoder with PyTorch

Primary LanguagePython

Course Project of CSE545 Advanced Computer Vision, UNIST

Context Encoder

Execution method

  1. Place your dataset in data/

  2. install requirement package

    pip install -r requirements.txt
  3. run the code

    python main.py

Description

Result

epoch 200, lambda 0.01