Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Rectangle training and training your own dataset #51

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

carlsummer
Copy link

Thank you very much for your sharing. I modified some of your code according to my own requirements, mainly to meet the following functions
After my own modification, I can train my own data set, and I can train rectangle, so I don't have to be square

@zhou13
Copy link
Owner

zhou13 commented Jul 8, 2021

Thanks for the PR! Have you tested your model and notice any performance improvement over image resizing?

@carlsummer
Copy link
Author

I tried 512 * 512 and 64 * 512, because the graph for my requirements is 400 * 3000, so it is better to test 64 * 512. The line appeared at the first verification, but it didn't appear until 512 * 512 for the third time

@carlsummer
Copy link
Author

I didn't compare the effects of different sizes on wireframe datasets. But you can try, training should not be too long, I because the company has the need to detect the line, so I changed the code to run rectangular; And I also hope that you can merge or open a new branch to help the readers

@carlsummer
Copy link
Author

image
image

@carlsummer
Copy link
Author

I saw several issues in the issue about how to train my own data set. The task my company now gives me is also to train my own data set, and then it is not a square picture. So I changed the code myself. Share with you, hope to help readers behind

@sachinkaundal
Copy link

sachinkaundal commented Jul 12, 2021

Hello @carlsummer, i modified train_test_split.py as per my requirements and trying to run but i am getting some error : ModuleNotFoundError: No module named 'python_developer_tools'
can you please help me to solve the issue. Thanks in advance for response.

@carlsummer
Copy link
Author

@sachinkaundal
https://github.com/carlsummer/python_developer_tools
That is a development tool I wrote. There are many tool functions in it. You can copy it,

@carlsummer
Copy link
Author

@sachinkaundal
I annotate it with label

@sachinkaundal
Copy link

sachinkaundal commented Jul 14, 2021 via email

@carlsummer
Copy link
Author

deploy onnx,and you can use onnxruntime run

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants