SketchNet

1. Environment configuration

• Ubuntu 16.04 (16GB)
• python3.5
• Tensorflow 1.1

2. Code:

DataGenTFRecord.py : From raw data generation TFRecord

Three Net Branch:

• SketchNet.py : sketch branch
  
• ImageNetNeg.py : Image Negative branch
  
• ImageNetPos.py : Image Positive branch
  

ReadData:

• ReadData_tfrecord.py : from TFRecord read data
  
• ReadData_np.py: Read Data in order
  
• ReadData_Random.py : Read Data in random
  

TrainModel:

• TrainModel_Siamese.py : Siamese architecture
  
• TrainModel_Heterogeneous.py : Heterogeneous architecture
  

3. Data:

DownLoad Link:( here )

4. References:

• Sketch Me That Shoe
  

Sketch	Image

Note:

• The picture has nothing to do with the content！