Train with my own dataset?

wangyuxin87 / ContourNet

A PyTorch implementation of "ContourNet: Taking a Further Step toward Accurate Arbitrary-shaped Scene Text Detection" (CVPR2020)

Other

227 stars 46 forks source link

Train with my own dataset? #7

Closed linkworb closed 4 years ago

linkworb commented 4 years ago

Can you tell me how to convert a dataset with only location annotations into COCO format? I want to train with my own dataset, but I don't konw how to convert it into COCO format.

He-haitao commented 4 years ago

Do you install ContourNet under torch 1.0, cudatoolkit 9.0? I met install problems with torch 1.5 cuda 10.0. When i run test_contour_net, it leads to many import problems about xx not exists.

wangyuxin87 commented 4 years ago

We upload a code to generate coco formate file from txt, which is avaliable at this link.