분류 전체보기
-
2개의 dataframe 합치기 (nan 없애고!)카테고리 없음 2022. 12. 8. 13:24
merge 대신 update를 쓰자 (초간단) https://stackoverflow.com/questions/59045507/how-to-merge-two-dataframes-and-keep-the-non-nan-values-in-it How to merge two dataframes and keep the non -nan values in it df1 0 Name phone Marks 1 mark 1225 20.0 2 charles 165498 36.5 3 oscar 1567 nan 4 bucky 13579 22.0 5 austin 1365... stackoverflow.com
-
[OCR] CRAFT 학습 간 주의사항컴퓨터 비젼(Computer Vision) 2022. 11. 27. 20:48
TL;DR 학습 데이터가 되는 입력 이미지의 모든 텍스트는 레이블링 하자. CRAFT란 naver clova에서 발표한 text detection 알고리즘으로 segmentation으로 text의 위치를 검출하는 알고리즘이다. 네트워크 구조는 UNet-like로, VGG를 backbone 구조로 사용하였다. 본 글에서 작성하고자 하는 내용은 비단 CRAFT 뿐만 아니라 대부분의 text detection 알고리즘에서 통용될 것이다. https://github.com/JaidedAI/EasyOCR/blob/054ec4f90127ab9e45f101baf8caa519d5a6e035/trainer/craft/loss/mseloss.py#L98는 CRAFT의 loss 함수이다. Loss 계산간 negative p..
-
-
Multi-gpu에서 학습된 모델의 weights를 Single-gpu (or CPU)에서 불러오기라이브러리/PyTorch 2022. 9. 12. 03:38
model = torchvision.models.segmentation.__dict__['deeplabv3_resnet50'](pretrained=False, aux_loss=False) device = torch.device("cpu") model.classifier[-1] = torch.nn.Conv2d(model.classifier[-1].in_channels, 1, kernel_size=model.classifier[-1].kernel_size) # change number of outputs to 1 model.to(device) # checkpoint = torch.load("output/segmentation/deeplabv3_resnet50_random/best.pt", map_locati..