LiamLYJ / scene_seg

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

scene_seg

pytorch implementation of scene_seg

pytorch >= 0.4

based on paper: One-shot Texture Segmentation https://arxiv.org/pdf/1807.02654.pdf

Train the model: python train.py (with args settings)

couple results are given in save_dtd(dtd dataset, http://www.robots.ox.ac.uk/~vgg/data/dtd/),
save_scene (use js script to download training dataset) and in real_test(real test results)

Test the model: python test.py (with args setting)

input image input texture output mask

input image input texture output mask

input image input texture output mask

About

License:MIT License


Languages

Language:Python 98.4%Language:JavaScript 1.6%