InChil2
commented
3 years ago Labelme 형식을 COCO 형식으로 변환 후 mAP 산출 시도 필요
Open pkwon35 opened 3 years ago
InChil2
commented
3 years ago Labelme 형식을 COCO 형식으로 변환 후 mAP 산출 시도 필요
/content/drive/MyDrive/Detection_of_lesions/code/Faster_RCNN/Faster_RCNN_불균등.ipynb 위 에서 학습 완료.
.pt 형식으로 학습한 모델 저장.
/content/drive/MyDrive/Detection_of_lesions/code/Faster_RCNN/model.pt
<< 모델 불러오는법 >>
num_classes = 5 model2 = get_instance_segmentation_model(num_classes) model2.to(device) model2.load_state_dict(torch.load(f'/content/drive/MyDrive/Detection_of_lesions/code/Faster_RCNN/model.pt'))