Re-implementation of basic papers in CV.
Clone the repository by
git clone https://github.com/HaojunYuPKU/Computer_Vision_Practice
, and install the requirement packages in your environment by
pip install -r requirements.txt
In the folder Wide_ResNet
, run
python main.py --depth [depth] --widen_factor [widen_factor] --gpu [gpus]
to train a Wide ResNet with certain depth and widen factor.
In the folder Relation_Net
, run
python main.py
to train a Relation Network.