Comments (2)
Hi @ys0923,
Thanks for the issue.
I think that there have been many queries for the dataset as well as a training example.
I had a very simple dataset of images with size 512x512x3 = (3, 512, 512) Tensor.
It had different types of animals in the dataset. Namely (Tiger, Lion, Cheetah and wolf)
The code is very generic it will work fine for any dataset folder.
The folder needs images labelled from 0, 1, 2, ... and so on.
Sadly I don't have a copy of dataset with me 😢 and hence I really cannot provide a kaggle kernel.
Can you try training on one maybe if you have sufficient bandwidth, more than 20 epochs or so?
Also see #5
The code has worked fine for many people. So I suppose it should satisfy your use case too.
Note that this model is very basic and really like an AlexNet, so no huge expectations for training / results.
from image_similarity.
https://www.kaggle.com/datasets/okeaditya/animals-data
from image_similarity.
Related Issues (8)
- problem in train code ,embedding = torch.cat((embedding, enc_output), 0)
- Why training autoencoder architecture different from inference? HOT 1
- Fine-tuning HOT 2
- Where can I get the animals dataset for the training script ? HOT 3
- UnboundLocalError: local variable 'loss' referenced before assignment HOT 7
- Animals dataset HOT 3
- Miss encoder checkpoint HOT 5
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from image_similarity.