Skip to content

Instantly share code, notes, and snippets.

View wkentaro's full-sized avatar

Kentaro Wada wkentaro

View GitHub Profile
@wkentaro
wkentaro / create_labelme_json.py
Created June 19, 2020 14:07
Create labelme compatible json file from image (random polygon)
import base64
import json
import os
import numpy as np
np.random.seed(0)
img_file = "lena.png"
import math
import imgviz
import numpy as np
import torch
class PositionalEmbeddingSinCos(torch.nn.Module):
def __init__(
self, num_pos_feats=64, temperature=10000, normalize=False, scale=None
from abc import ABC, abstractmethod
from typing import Any, List
import warnings
import numpy as np
class Summary(object):
def __init__(self, name: str, value: Any):
self.name = name
import imgviz
import tqdm
import corvus_segmentation_models
dataset = corvus_segmentation_models.datasets.Warehouse20200526Dataset(
split="sim"
)
#!/usr/bin/env python
import json
import imgviz
import numpy as np
import path
import trimesh
import trimesh.transformations as tf
#!/usr/bin/env python
train_pix_mal = 67990163
train_pix_ben = 31488044
train_pix_nor = 38122542
train_pix_out = 1466996531
train_pix_all = train_pix_mal + train_pix_ben + train_pix_nor + train_pix_out
valid_pix_mal = 6431428
#!/usr/bin/env python
import numpy as np
import torch
import corvus_segmentation_models
from train import TransformDataset
#!/bin/bash
convert 2004.04336.pdf -thumbnail x256 thumb.png
imtile *.png --shape 1x10 --out thumb.png
#!/usr/bin/env python
import gdown
import numpy as np
import scipy.io
import corvus_segmentation_models
mat_file = gdown.cached_download(
#!/usr/bin/env python
import json
import numpy as np
import path
import trimesh
import morefusion