FiNet-pytorch

data_crawler: web crawler using scrapy
Fi_dataset.py: data loader
train.py: train the network
test.py: test the network
models.py: model framework
utils.py: some tool functions

A pytorch re-implementation of "FiNet: Compatible and Diverse Fashion Image Inpainting"

Code structure

This re-implementation has been based on this paper and tested with pytorch 0.4.1 on Ubuntu 16.04(GeForce GTX 1080 Ti).

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
data_crawler		data_crawler
data_preprocess		data_preprocess
.DS_Store		.DS_Store
Fi_dataset.py		Fi_dataset.py
README.md		README.md
memo.txt		memo.txt
models.py		models.py
test.py		test.py
train.py		train.py
utils.py		utils.py