- Ubuntu 16.04 (16GB)
- python3.5
- Tensorflow 1.1
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
DownLoad Link:( here )
| Sketch | Image |
|---|---|
 |
 |
Note:
- The picture has nothing to do with the content！