├── .gitignore ├── README.md ├── classification └── train.py ├── data ├── dataset-15 │ └── dataset.csv ├── extracted_meta.csv └── parts_category_list.csv ├── dataset ├── blender │ ├── configs │ │ ├── augmentation.json │ │ └── thumbnail.json │ ├── render.py │ ├── sphere.py │ └── utils.py ├── color_viewer.py ├── crop.py ├── generate_dataset.py ├── generate_thumbnails.py └── utils.py ├── environment.yml └── examples ├── 6538b-top_images_ssim.png ├── 6538b.svg ├── category_distribution.svg ├── cm_resnext50_freezed.svg ├── loss_resnext50_freezed.svg ├── rendered_brick2_noise.jpg └── rendered_brick_noise.jpg /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/README.md -------------------------------------------------------------------------------- /classification/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/classification/train.py -------------------------------------------------------------------------------- /data/dataset-15/dataset.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/data/dataset-15/dataset.csv -------------------------------------------------------------------------------- /data/extracted_meta.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/data/extracted_meta.csv -------------------------------------------------------------------------------- /data/parts_category_list.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/data/parts_category_list.csv -------------------------------------------------------------------------------- /dataset/blender/configs/augmentation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/blender/configs/augmentation.json -------------------------------------------------------------------------------- /dataset/blender/configs/thumbnail.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/blender/configs/thumbnail.json -------------------------------------------------------------------------------- /dataset/blender/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/blender/render.py -------------------------------------------------------------------------------- /dataset/blender/sphere.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/blender/sphere.py -------------------------------------------------------------------------------- /dataset/blender/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/blender/utils.py -------------------------------------------------------------------------------- /dataset/color_viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/color_viewer.py -------------------------------------------------------------------------------- /dataset/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/crop.py -------------------------------------------------------------------------------- /dataset/generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/generate_dataset.py -------------------------------------------------------------------------------- /dataset/generate_thumbnails.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/generate_thumbnails.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/environment.yml -------------------------------------------------------------------------------- /examples/6538b-top_images_ssim.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/6538b-top_images_ssim.png -------------------------------------------------------------------------------- /examples/6538b.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/6538b.svg -------------------------------------------------------------------------------- /examples/category_distribution.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/category_distribution.svg -------------------------------------------------------------------------------- /examples/cm_resnext50_freezed.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/cm_resnext50_freezed.svg -------------------------------------------------------------------------------- /examples/loss_resnext50_freezed.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/loss_resnext50_freezed.svg -------------------------------------------------------------------------------- /examples/rendered_brick2_noise.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/rendered_brick2_noise.jpg -------------------------------------------------------------------------------- /examples/rendered_brick_noise.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jtheiner/LegoBrickClassification/HEAD/examples/rendered_brick_noise.jpg --------------------------------------------------------------------------------