Greg Chu gregchu
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| Contrastive Learning Script | |
| Aims: | |
| - Train model to move embeddings closer that are of the same class | |
| - Visualize embeddings in 2D space before and after contrastive learning | |
| - Calculate and track intraclass variance metrics | |
| """ | |
| import os |
gregchu
/ encode-decode-jpeg
Created
October 2, 2017 21:16
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from StringIO import StringIO | |
| from PIL import Image | |
| def encode_jpeg(img, quality=100): | |
| """Encode ndarray image into a jpeg | |
| Args: | |
| img: 3D (wxhxRGB) array | |
| quality: jpeg compression quality | |
| """ | |
| s = StringIO() |
gregchu
/ predict.py
Created
March 10, 2017 06:51
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| def predict(model, img, target_size, top_n=3): | |
| """Run model prediction on image | |
| Args: | |
| model: keras model | |
| img: PIL format image | |
| target_size: (w,h) tuple | |
| top_n: # of top predictions to return | |
| Returns: | |
| np array of predictions | |
| """ |
gregchu
/ ImageNet ILSVRC labels short list
Created
March 10, 2017 04:48
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| dog | |
| bear | |
| chair | |
| car | |
| keyboard | |
| crate | |
| crib | |
| flagpole | |
| iPod | |
| boat |
gregchu
/ PASCAL VOC labels
Created
February 8, 2017 17:30
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Person | |
| Car | |
| Bicycle | |
| Bus | |
| Motorbike | |
| Train | |
| Aeroplane | |
| Chair | |
| Bottle | |
| Dining Table |
gregchu
/ ImageNet ILSVRC labels
Last active
November 10, 2025 01:39
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| tench, Tinca tinca | |
| goldfish, Carassius auratus | |
| great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias | |
| tiger shark, Galeocerdo cuvieri | |
| hammerhead, hammerhead shark | |
| electric ray, crampfish, numbfish, torpedo | |
| stingray | |
| cock | |
| hen | |
| ostrich, Struthio camelus |
gregchu
/ COCO labels
Last active
November 29, 2018 12:56
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| person | |
| bicycle | |
| car | |
| motorcycle | |
| airplane | |
| bus | |
| train | |
| truck | |
| boat | |
| traffic light |