https://github.com/knight-x/caffe2-vision
Caffe2-vision help engineer to build vision CNN for training and product, including dataset maker and favorite models
https://github.com/knight-x/caffe2-vision
caffe2 cnn compter-vision vision
Last synced: about 1 month ago
JSON representation
Caffe2-vision help engineer to build vision CNN for training and product, including dataset maker and favorite models
- Host: GitHub
- URL: https://github.com/knight-x/caffe2-vision
- Owner: Knight-X
- Created: 2017-07-24T09:15:53.000Z (almost 9 years ago)
- Default Branch: master
- Last Pushed: 2017-08-02T15:30:15.000Z (almost 9 years ago)
- Last Synced: 2025-01-05T21:29:49.482Z (over 1 year ago)
- Topics: caffe2, cnn, compter-vision, vision
- Language: Python
- Homepage:
- Size: 4.88 KB
- Stars: 2
- Watchers: 2
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# caffe2-vision
This is the util for making image dataset for Caffe2 .
## Datasets Module:
It will search and load images from folder, and help to create the dataset easily .
* ImageFolder
* ImageLoader
### ImageFolder
```
rootfolder/boat/xxx.jpg
rootfolder/boat/123.jpg
rootfolder/boat/456.jpg
rootfolder/car/123.jpg
rootfolder/car/456.jpg
rootfolder/car/cjf.jpg
```
```
import imgfolder as folder
import imgloader as loader
dataset = folder.ImageFolder("your file path")
imgs = loader.ImgLoader(dataset)
```
##### imgs is the list contains (index,images(img_data, label_data))
```
for i, data in enumerate(imgs):
img_data, label_data = data
```
## DB Writer
- minidb
## Models
- ResNet
This is inspired by pytorch/vision.