[Question] How to classify designs?

Cristobal_Del_Piero: <p>Hello, I’m fairly new to this world of vision. But I want to train a network that tells me whether it’s product 1 or 2. The products are quite similar, and I’ve trained with many photos of these designs, but I always get disastrous results. Can someone help me or recommend a way to detect one product from another? I upload photos of the objects I’m trying to detect.<br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/b/1/b1617431bf273bd49eddc5d79cee50748588125d.jpeg" data-download-href="https://discuss.pytorch.org/uploads/default/b1617431bf273bd49eddc5d79cee50748588125d" title="93a449d4-7ee4-47bf-a679-0af66d2c23cd"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/b/1/b1617431bf273bd49eddc5d79cee50748588125d_2_666x500.jpeg" alt="93a449d4-7ee4-47bf-a679-0af66d2c23cd" data-base62-sha1="pjbiZeHErhKnUgxJ59llB6Cy6Nn" width="666" height="500" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/b/1/b1617431bf273bd49eddc5d79cee50748588125d_2_666x500.jpeg, https://discuss.pytorch.org/uploads/default/optimized/3X/b/1/b1617431bf273bd49eddc5d79cee50748588125d_2_999x750.jpeg 1.5x, https://discuss.pytorch.org/uploads/default/optimized/3X/b/1/b1617431bf273bd49eddc5d79cee50748588125d_2_1332x1000.jpeg 2x" data-dominant-color="665339"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">93a449d4-7ee4-47bf-a679-0af66d2c23cd</span><span class="informations">1600×1200 168 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>

J_Johnson: <p><a class="mention" href="/u/cristobal_del_piero">@Cristobal_Del_Piero</a> Welcome to the forums!</p>
<p>Can you share your model and training code?</p>
<p>Also, what is the ratio of images in your dataset for your two classes?</p>

Cristobal_Del_Piero: <p>Thank you for your response. So far, I’ve tried using YOLO with a database, but I’ve found that it’s not the best option. As I mentioned earlier, I’m new to this vision world and learning on my own. I’ve opted for PyTorch to try to train a neural network for its ease of installation with my hardware.</p>
<p>So far, the only thing that has worked for me is using Brute Force and ORB to “try” to identify one object or another, but that’s not deep learning, and the truth is, I don’t know where to start. All I’m looking for is some guidance on which path to follow.</p>

J_Johnson: <p>Here is a simple PyTorch tutorial on classification problems: <a href="https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html" class="inline-onebox" rel="noopener nofollow ugc">Training a Classifier — PyTorch Tutorials 2.1.0+cu121 documentation</a></p>
<p>Although in your actual case, you may want to use a more advanced model, like a Resnet with transfer learning:<br>
<a href="https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html" class="onebox" target="_blank" rel="noopener nofollow ugc">https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html</a></p>